Merge pull request #253 from kvdotkirti/kvdotkirti-patch-1

Partial: Update pytorch load fail from print to logging #251 Incomplete - logging.error will be the level of the root logger for successive logging calls.
icenet-ai · May 29, 2024 · 3206a74 · 3206a74
2 parents b03f03f + b29a636
commit 3206a74
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/icenet/data/dataset.py b/icenet/data/dataset.py
@@ -20,9 +20,9 @@
     from torch.utils.data import Dataset
     pytorch_available = True
 except ModuleNotFoundError:
-    print("PyTorch not found - not required if not using PyTorch")
+    logger.error("PyTorch not found - not required if not using PyTorch")
 except ImportError:
-    print("PyTorch import failed - not required if not using PyTorch")
+    logger.error("PyTorch import failed - not required if not using PyTorch")
 
 """