replace RuntimeWarning by logging

thorben-frank · Apr 22, 2024 · 706fcf6 · 706fcf6
1 parent 04ebed8
commit 706fcf6
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/mlff/data/dataloader_sparse_tfds.py b/mlff/data/dataloader_sparse_tfds.py
@@ -8,15 +8,21 @@
 try:
     import tensorflow as tf
 except ModuleNotFoundError:
-    raise RuntimeWarning(
+    logging.warning(
         "For using TFDSDataLoader please install tensorflow."
     )
+    # raise RuntimeWarning(
+    #     "For using TFDSDataLoader please install tensorflow."
+    # )
 try:
     import tensorflow_datasets as tfds
 except ModuleNotFoundError:
-    raise RuntimeWarning(
+    logging.warning(
         "For using TFDSDataLoader please install tensorflow_datasets."
     )
+    # raise RuntimeWarning(
+    #     "For using TFDSDataLoader please install tensorflow_datasets."
+    # )
 
 
 logging.MLFF = 35

diff --git a/mlff/data/dataloader_sparse_tfrecord.py b/mlff/data/dataloader_sparse_tfrecord.py
@@ -11,9 +11,12 @@
 try:
     import tensorflow as tf
 except ModuleNotFoundError:
-    raise RuntimeWarning(
-        "For using TFRecordDataLoaderSparse please install tensorflow."
+    logging.warning(
+        "For using TFDSDataLoader please install tensorflow."
     )
+    # raise RuntimeWarning(
+    #     "For using TFRecordDataLoaderSparse please install tensorflow."
+    # )
 
 logging.MLFF = 35
 logging.addLevelName(logging.MLFF, 'MLFF')