From 0aebea6421d6730ca98e4d20e5aab0d6f3227cfd Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Wed, 3 Apr 2024 09:01:32 +0200
Subject: [PATCH] [utils] Fix import
---
src/mednet/scripts/utils.py | 9 ++++++---
1 file changed, 6 insertions(+), 3 deletions(-)
diff --git a/src/mednet/scripts/utils.py b/src/mednet/scripts/utils.py
index b1adfb93..4c894bd7 100644
--- a/src/mednet/scripts/utils.py
+++ b/src/mednet/scripts/utils.py
@@ -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())
--
GitLab