SchNet threahsold increased for conv_head

ORNL · Dec 13, 2024 · e0e7fb6 · e0e7fb6
1 parent 1a39140
commit e0e7fb6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_graphs.py b/tests/test_graphs.py
@@ -164,6 +164,7 @@ def unittest_train_model(
         thresholds["PNAPlus"] = [0.2, 0.15]
     if ci_input == "ci_conv_head.json":
         thresholds["GIN"] = [0.25, 0.40]
+        thresholds["SchNet"] = [0.30, 0.30]
 
     verbosity = 2