Minor model test fix

aqlaboratory · Aug 10, 2023 · 3de188e · 3de188e
1 parent a18f98c
commit 3de188e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -55,7 +55,7 @@ def test_dry_run(self):
  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