Fix model test

aqlaboratory · Jun 24, 2023 · d5da89c · d5da89c
1 parent df06706
commit d5da89c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -48,6 +48,7 @@ def test_dry_run(self):
         # deepspeed for this test
 
         model = AlphaFold(c)
+        model.eval()
 
         batch = {}
         tf = torch.randint(c.model.input_embedder.tf_dim - 1, size=(n_res,))