changed threshold in PNA test with convolutional heads

ORNL · Jan 19, 2024 · 3715acf · 3715acf
1 parent 310089b
commit 3715acf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_graphs.py b/tests/test_graphs.py
@@ -140,6 +140,7 @@ def unittest_train_model(model_type, ci_input, use_lengths, overwrite_data=False
     if use_lengths and "vector" in ci_input:
         thresholds["PNA"] = [0.2, 0.15]
     if ci_input == "ci_conv_head.json":
+        thresholds["PNA"] = [0.5, 0.5]
         thresholds["EGNN"] = [0.5, 0.5]
         thresholds["SchNet"] = [0.5, 0.5]
     verbosity = 2