Merge pull request #299 from Paperspace/PS-14050-Fix_NotImplementedError_when_parsing_some_experiments

Fix NotImplementedError when parsing experiments with volume options

Author: BartoszCki

gradient/api_sdk/models/__init__.py

@@ -1,6 +1,6 @@
 from .artifact import Artifact
 from .cluster import Cluster
-from .dataset import Dataset
+from .dataset import Dataset, VolumeOptions
 from .deployment import Deployment
 from .experiment import BaseExperiment, MultiNodeExperiment, SingleNodeExperiment, MpiMultiNodeExperiment
 from .hyperparameter import Hyperparameter

gradient/api_sdk/serializers/dataset.py

@@ -6,6 +6,8 @@
 
 
 class DatasetVolumeOptionsSchema(BaseSchema):
+    MODEL = models.VolumeOptions
+
     kind = ma.fields.Str(required=True)
     size = ma.fields.Str(required=True)
 

tests/example_responses.py

@@ -36,7 +36,16 @@
                 },
                 "triggerEvent": None,
                 "triggerEventId": None
-            }
+            },
+            "datasets": [
+                {
+                    "uri": "s3://some/dataset",
+                    "volume_options": {
+                        "kind": "dynamic",
+                        "size": "10Gi",
+                    },
+                },
+            ],
         },
         {
             "dtCreated": "2019-03-21T07:46:57.706055+00:00",
@@ -4586,7 +4595,16 @@
             },
             "triggerEvent": None,
             "triggerEventId": None
-        }
+        },
+        "datasets": [
+            {
+                "uri": "s3://some/dataset",
+                "volume_options": {
+                    "kind": "dynamic",
+                    "size": "10Gi",
+                },
+            },
+        ],
     },
     "message": "success"
 }