Make triangle attention test more sensitive

80f968fe · Gustaf Ahdritz · de91f011 · 80f968fe
Commit 80f968fe 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)
+        pair_act = np.random.rand(n_res, n_res, consts.c_z) * 10
        pair_mask = np.random.randint(low=0, high=2, size=(n_res, n_res))

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