fix new quality errors (#1156)

neuralmagic · Jul 31, 2023 · 49a23e7 · 49a23e7
1 parent 2c9035e
commit 49a23e7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/deepsparse/utils/data.py b/src/deepsparse/utils/data.py
@@ -101,7 +101,7 @@ def verify_outputs(
             raise Exception(
                 f"Output shapes don't match, {output.shape} != {gt_output.shape}"
             )
-        if type(output) != type(gt_output):
+        if type(output) is not type(gt_output):
             raise Exception(
                 f"Output types don't match, {type(output)} != {type(gt_output)}"
             )

diff --git a/tests/deepsparse/pipelines/test_dynamic_batch_pipeline.py b/tests/deepsparse/pipelines/test_dynamic_batch_pipeline.py
@@ -35,7 +35,7 @@
 
 
 def compare(expected, actual):
-    assert type(expected) == type(actual)
+    assert type(expected) is type(actual)
 
     if isinstance(expected, (list, float, numpy.ndarray)):
         expected_np = numpy.asarray(expected, dtype=float)