3 files changed, 105 insertions, 0 deletions
diff --git a/spec/fixtures/api/schemas/ml/get_experiment.json b/spec/fixtures/api/schemas/ml/get_experiment.json
new file mode 100644
index 00000000000..cf8da7f999f
--- /dev/null
+++ b/spec/fixtures/api/schemas/ml/get_experiment.json
@@ -0,0 +1,23 @@
+{
+  "type": "object",
+  "required": [
+    "experiment"
+  ],
+  "properties": {
+    "experiment": {
+      "type": "object",
+      "required" : [
+        "experiment_id",
+        "name",
+        "artifact_location",
+        "lifecycle_stage"
+      ],
+      "properties" : {
+        "experiment_id": { "type": "string" },
+        "name": { "type": "string" },
+        "artifact_location": { "type": "string" },
+        "lifecycle_stage": { "type": { "enum" : ["active", "deleted"] } }
+      }
+    }
+  }
+}
diff --git a/spec/fixtures/api/schemas/ml/run.json b/spec/fixtures/api/schemas/ml/run.json
new file mode 100644
index 00000000000..2418f44b21f
--- /dev/null
+++ b/spec/fixtures/api/schemas/ml/run.json
@@ -0,0 +1,47 @@
+{
+  "type": "object",
+  "required": [
+    "run"
+  ],
+  "properties": {
+    "run": {
+      "type": "object",
+      "required": [
+        "info",
+        "data"
+      ],
+      "properties": {
+        "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": "",
+            "status": { "type": { "enum" : ["RUNNING", "SCHEDULED", "FINISHED", "FAILED", "KILLED"] } },
+            "lifecycle_stage": { "type": { "enum" : ["active"] } }
+          }
+        },
+        "data": {
+          "type": "object"
+        }
+      }
+    }
+  }
+}
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"] } }
+      }
+    }
+  }
+}