Minor model test fix

3de188e9 · Christina Floristean · a18f98cf · 3de188e9
Commit 3de188e9 authored Aug 10, 2023 by Christina Floristean
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_model.py tests/test_model.py +1 -1

No files found.
--- a/tests/test_model.py
+++ b/tests/test_model.py
@@ -55,7 +55,7 @@ class TestModel(unittest.TestCase):
        n_res = consts.n_res
        n_extra_seq = consts.n_extra

-        c = model_config(consts.model, train=True)
+        c = model_config(consts.model)
        c.model.evoformer_stack.no_blocks = 4  # no need to go overboard here
        c.model.evoformer_stack.blocks_per_ckpt = None  # don't want to set up
        # deepspeed for this test