Fix FP precision issue in test

8048efd6 · Gustaf Ahdritz · 4dd50722 · 8048efd6
Commit 8048efd6 authored Oct 08, 2021 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_structure_module.py tests/test_structure_module.py +1 -1

No files found.
--- a/tests/test_structure_module.py
+++ b/tests/test_structure_module.py
@@ -141,7 +141,7 @@ class TestStructureModule(unittest.TestCase):
        }
        batch = {
-            'seq_mask': np.random.randint(0, 2, (n_res,)),
+            'seq_mask': np.random.randint(0, 2, (n_res,)).astype(np.float32),
            'aatype': np.random.randint(0, 21, (n_res,)),
        }