update codegen test

2 years ago · a005965d2d
parent 3abbaf8bc6
commit a005965d2d
1 changed files with 1 additions and 1 deletions
--- a/tests/test_autochunk/test_autochunk_codegen.py
+++ b/tests/test_autochunk/test_autochunk_codegen.py
@ -99,7 +99,7 @@ def _test_autochunk_codegen(rank, msa_len, pair_len, max_memory):
    gpc.destroy()


-@pytest.mark.parametrize("max_memory", [None, 20, 24, 28, 32])
+@pytest.mark.parametrize("max_memory", [None, 20, 25, 30])
@pytest.mark.parametrize("msa_len", [32])
@pytest.mark.parametrize("pair_len", [64])
 def test_autochunk_codegen(msa_len, pair_len, max_memory):