Raise exception when we fail to detect the model's framework

waleedka · Nov 28, 2018 · 28d791e · 28d791e
1 parent 2daeadf
commit 28d791e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/hiddenlayer/graph.py b/hiddenlayer/graph.py
@@ -142,7 +142,9 @@ def build_graph(model=None, args=None, input_names=None,
     elif framework == "tensorflow":
         from .tf_builder import import_graph, FRAMEWORK_TRANSFORMS
         import_graph(g, model)
-
+    else:
+        raise ValueError("`model` input param must be a PyTorch, TensorFlow, or Keras-with-TensorFlow-backend model.") 
+
     # Apply Transforms
     if framework_transforms:
         if framework_transforms == "default":