We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a2bdc85 commit 825c2dbCopy full SHA for 825c2db
1 file changed
sub-packages/bionemo-evo2/tests/bionemo/evo2/run/test_predict.py
@@ -352,6 +352,8 @@ def test_predict_evo2_runs_with_log_probs(
352
assert len(seq_idx_map) == num_sequences
353
for original_idx, log_probs in zip(preds["seq_idx"], preds["log_probs_seqs"]):
354
if mp_size > 1 or fp8:
355
+ # FIXME changing batch size so it doesn't match also required dropping rel=1e-6 to rel=1e-3.
356
+ # This should be investigated.
357
rel = 1e-3
358
else:
359
rel = 1e-6
0 commit comments