Skip to content

Commit 1104caa

Browse files
committed
one last fix for cpu gpu test
1 parent 437fee3 commit 1104caa

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

metapredict/tests/test_metapredict_CPU_GPU.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -259,8 +259,8 @@ def test_pLDDT_v1_cpu_vs_gpu(sequences=sequences):
259259
cur_gpu_scores = gpu_scores[seq_name][1]
260260
for i in range(len(cur_cpu_scores)):
261261
# plddt V1 scores are 100x bigger than plldt V2 (which are trained on plddt / 100)
262-
# so we want to use 0.01 here instead of 0.001
263-
assert close_enough(cur_cpu_scores[i], cur_gpu_scores[i], allowed_error=0.02)==True
262+
# so we want to use 0.1 here instead of 0.001
263+
assert close_enough(cur_cpu_scores[i], cur_gpu_scores[i], allowed_error=0.1)==True
264264

265265
def test_pLDDT_v2_cpu_vs_gpu(sequences=sequences):
266266
version='v2'

0 commit comments

Comments
 (0)