fix: library is now working without tensorflow

MLRichter · Jan 14, 2022 · 588c2d1 · 588c2d1
1 parent d5eb995
commit 588c2d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rfa_toolbox/__init__.py b/rfa_toolbox/__init__.py
@@ -15,7 +15,7 @@ def create_graph_from_pytorch_model(*args, **kwargs):
     from rfa_toolbox.encodings.tensorflow_keras.ingest_architecture import (
         create_graph_from_model as create_graph_from_tensorflow_model,
     )
-except ValueError:
+except ImportError:
 
     def create_graph_from_tensorflow_model(*args, **kwargs):
         raise ImportError("This function is not available, tensorflow not installed")