From b29a6364124622add7aec4388809e8b71fb616ac Mon Sep 17 00:00:00 2001
From: kvdotkirti <152726212+kvdotkirti@users.noreply.github.com>
Date: Sat, 13 Apr 2024 10:46:57 +0530
Subject: [PATCH] Update dataset.py

---
 icenet/data/dataset.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/icenet/data/dataset.py b/icenet/data/dataset.py
index 8763697..5eca581 100644
--- a/icenet/data/dataset.py
+++ b/icenet/data/dataset.py
@@ -19,9 +19,9 @@
 try:
     from torch.utils.data import Dataset
 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")
 
 """