fix layer tests (#577)

obss · Aug 29, 2022 · cfdd7f6 · cfdd7f6
1 parent 072ea62
commit cfdd7f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_layer.py b/tests/test_layer.py
@@ -32,7 +32,7 @@ def test_load_layer_model(self, mock_layer_get_model):
             # Run assertions
             mock_layer_get_model.assert_called_once_with(name=layer_model_path, no_cache=False)
 
-            self.assertIsInstance(detection_model, type(Yolov5DetectionModel()))
+            self.assertIsInstance(detection_model, Yolov5DetectionModel)
             self.assertIsInstance(detection_model.model, type(yolo_model))
 
         @mock.patch("layer.get_model")