Added missing parameters to the schema

metatensor · Sep 24, 2024 · 8a51cb2 · 8a51cb2
1 parent 4d4b76c
commit 8a51cb2
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/metatrain/experimental/pet/schema-hypers.json b/src/metatrain/experimental/pet/schema-hypers.json
@@ -101,6 +101,20 @@
             }
           ]
         },
+        "K_CUT_DELTA": {
+          "oneOf": [
+            {
+              "type": "number"
+            },
+            {
+              "type": "null"
+            }
+          ]
+        },
+        "DTYPE": {
+          "type": "string",
+          "enum": ["float32", "float64", "bfloat16"]
+        },
         "N_TARGETS": {
           "type": "integer"
         },