flaking

esa · Oct 10, 2023 · c6baf39 · c6baf39
1 parent fa051f4
commit c6baf39
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchquad/tests/integration_test_functions.py b/torchquad/tests/integration_test_functions.py
@@ -44,7 +44,7 @@ def __init__(
         """
         self.integration_dim = integration_dim
         self.expected_result = expected_result
-        if type(integrand_dims) == int or hasattr(integrand_dims, "__len__"):
+        if type(integrand_dims) is int or hasattr(integrand_dims, "__len__"):
             self.integrand_dims = integrand_dims
         else:
             ValueError(