diff options
Diffstat (limited to 'tests/test_eval_classification.py')
| -rw-r--r-- | tests/test_eval_classification.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/test_eval_classification.py b/tests/test_eval_classification.py index 3f9994d..7114166 100644 --- a/tests/test_eval_classification.py +++ b/tests/test_eval_classification.py @@ -45,7 +45,7 @@ class StubModel(torch.nn.Module): def _sieve35(n): c = n + 1 - while any(c % d == 0 for d in (2, 3, 5, 7)): + while any(c % d == 0 for d in (2, 3, 5, 7) if d < c): c += 1 return c |