Fix for python 3.7

Xilinx · Sep 12, 2023 · 7b2b36a · 7b2b36a
1 parent 713a778
commit 7b2b36a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -2,10 +2,13 @@
 # SPDX-License-Identifier: BSD-3-Clause
 
 import hypothesis
+from packaging import version
 import torch
 
 SEED = 123456
 DIFFERING_EXECUTOR_ENUM = 10  # Disabled since it fails with normal test setup
+HYPOTHESIS_HEALTHCHECK_VERSION = '6.83.2'  # The new healthcheck was introduced in this version
 
 torch.random.manual_seed(SEED)
-hypothesis.settings(suppress_health_check=[hypothesis.HealthCheck(DIFFERING_EXECUTOR_ENUM)])
+if version.parse(hypothesis.__version__) >= version.parse(HYPOTHESIS_HEALTHCHECK_VERSION):
+    hypothesis.settings(suppress_health_check=[hypothesis.HealthCheck(DIFFERING_EXECUTOR_ENUM)])