Fix model test

d5da89c1 · Gustaf Ahdritz · df067065 · d5da89c1
Commit d5da89c1 authored Jun 24, 2023 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

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

No files found.
--- a/tests/test_model.py
+++ b/tests/test_model.py
@@ -48,6 +48,7 @@ class TestModel(unittest.TestCase):
        # deepspeed for this test

        model = AlphaFold(c)
+        model.eval()

        batch = {}
        tf = torch.randint(c.model.input_embedder.tf_dim - 1, size=(n_res,))