1 files changed, 35 insertions, 0 deletions
diff --git a/spec/fixtures/api/schemas/ml/update_run.json b/spec/fixtures/api/schemas/ml/update_run.json
new file mode 100644
index 00000000000..b429444120f
--- /dev/null
+++ b/spec/fixtures/api/schemas/ml/update_run.json
@@ -0,0 +1,35 @@
+{
+  "type": "object",
+  "required": [
+    "run_info"
+  ],
+  "properties": {
+    "run_info": {
+      "type": "object",
+      "required": [
+        "run_id",
+        "run_uuid",
+        "user_id",
+        "experiment_id",
+        "status",
+        "start_time",
+        "artifact_uri",
+        "lifecycle_stage"
+      ],
+      "optional": [
+        "end_time"
+      ],
+      "properties": {
+        "run_id": { "type": "string" },
+        "run_uuid": { "type": "string" },
+        "experiment_id": { "type": "string" },
+        "artifact_location": { "type": "string" },
+        "start_time": { "type": "integer" },
+        "end_time": { "type": "integer" },
+        "user_id": { "type": "string" },
+        "status": { "type": { "enum" : ["RUNNING", "SCHEDULED", "FINISHED", "FAILED", "KILLED"] } },
+        "lifecycle_stage": { "type": { "enum" : ["active"] } }
+      }
+    }
+  }
+}