From cfdd7f6a824363e64320befdc8fe49c9cd24e45c Mon Sep 17 00:00:00 2001 From: fatih <34196005+fcakyon@users.noreply.github.com> Date: Tue, 30 Aug 2022 01:58:48 +0300 Subject: [PATCH] fix layer tests (#577) --- tests/test_layer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_layer.py b/tests/test_layer.py index 43a9fb4ff..ada37b3df 100644 --- 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")