diff --git a/tests/custom_op/test_multi_thresholding.py b/tests/custom_op/test_multi_thresholding.py index c757ee96b80c26833554126b399127f0b3467d78..5fc69ce50062d5366dec7c4dce3d87e4bd2bacf5 100644 --- a/tests/custom_op/test_multi_thresholding.py +++ b/tests/custom_op/test_multi_thresholding.py @@ -295,6 +295,7 @@ def test_execute_multi_thresholding(): assert (results_scaled == outputs_scaled).all() # performance and random test + np.random.seed(0) inputs = np.random.random((1, 256, 64, 64)) thresholds = (np.array([[1, 2, 3, 4, 5, 6]]) - 0.5) / 6