Make triangular attention test even more sensitive

6c5cd945 · Gustaf Ahdritz · 80f968fe · 6c5cd945
Commit 6c5cd945 authored Dec 10, 2021 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_triangular_attention.py tests/test_triangular_attention.py +1 -1

No files found.
--- a/tests/test_triangular_attention.py
+++ b/tests/test_triangular_attention.py
@@ -70,7 +70,7 @@ class TestTriangularAttention(unittest.TestCase):

        n_res = consts.n_res

-        pair_act = np.random.rand(n_res, n_res, consts.c_z) * 10
+        pair_act = np.random.rand(n_res, n_res, consts.c_z) * 100
        pair_mask = np.random.randint(low=0, high=2, size=(n_res, n_res))

        # Fetch pretrained parameters (but only from one block)]