Skip to content
Snippets Groups Projects
Commit 0aebea64 authored by Daniel CARRON's avatar Daniel CARRON :b: Committed by André Anjos
Browse files

[utils] Fix import

parent 901c950a
No related branches found
No related tags found
1 merge request!46Create common library
......@@ -12,8 +12,7 @@ import shutil
import lightning.pytorch
import lightning.pytorch.callbacks
import torch.nn
from ..engine.device import SupportedPytorchDevice
from medbase.engine.device import SupportedPytorchDevice
logger = logging.getLogger(__name__)
......@@ -59,7 +58,11 @@ def device_properties(
Static properties of the current machine.
"""
from ..utils.resources import cpu_constants, cuda_constants, mps_constants
from medbase.utils.resources import (
cpu_constants,
cuda_constants,
mps_constants,
)
retval: dict[str, int | float | str] = {}
retval.update(cpu_constants())
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment