Merge pull request #294 from Paperspace/PS-13914-Add_deployment_state_constants

Ps 13914 add deployment state constants

Author: BartoszCki

gradient/api_sdk/constants.py

@@ -19,7 +19,7 @@ def get_type_str(cls, state_int):
 
 
 class ExperimentState:
-    CREATED = 1
+    PENDING = 1
     PROVISIONED = 2
     NETWORK_SETUP = 3
     RUNNING = 4
@@ -28,15 +28,16 @@ class ExperimentState:
     FAILED = 7
     CANCELLED = 8
     NETWORK_TEARDOWN = 9
-    PENDING = 10
+    CREATED = 10
     PROVISIONING = 11
     NETWORK_SETTING_UP = 12
     NETWORK_TEARING_DOWN = 13
+    ABORTING = 14
 
     @classmethod
     def get_state_str(cls, state_int):
         state_strings = {
-            1: "created",
+            1: "pending",
             2: "provisioned",
             3: "network setup",
             4: "running",
@@ -45,7 +46,7 @@ def get_state_str(cls, state_int):
             7: "failed",
             8: "canceled",
             9: "network teardown",
-            10: "pending",
+            10: "created",
             11: "provisioning",
             12: "network setting up",
             13: "network tearing down",
@@ -149,3 +150,26 @@ class DatasetVolumeKinds:
         ("shared", DatasetVolumeKinds.SHARED),
     ),
 )
+
+
+class DeploymentState:
+    BUILDING = 1
+    PROVISIONING = 2
+    STARTING = 3
+    RUNNING = 4
+    STOPPING = 5
+    STOPPED = 6
+    ERROR = 7
+
+    @classmethod
+    def get_state_str(cls, state_int):
+        state_strings = {
+            1: "building",
+            2: "provisioning",
+            3: "starting",
+            4: "running",
+            5: "stopping",
+            6: "stopped",
+            7: "error",
+        }
+        return state_strings.get(state_int, "undefined")

tests/functional/test_experiments.py

@@ -927,7 +927,7 @@ class TestExperimentDetail(object):
 | Name                | some_name                |
 +---------------------+--------------------------+
 | ID                  | emarbao6t6tsn            |
-| State               | pending                  |
+| State               | created                  |
 | Artifact directory  | /some/artifact/directory |
 | Cluster ID          | clqr4b0ox                |
 | Experiment Env      | {'key': 'value'}         |
@@ -953,7 +953,7 @@ class TestExperimentDetail(object):
 | Name                | dsfads         |
 +---------------------+----------------+
 | ID                  | esro6mbmiulvbl |
-| State               | created        |
+| State               | pending        |
 | Ports               | 5000           |
 | Project ID          | prq70zy79      |
 | Worker Command      | sadas          |
@@ -1060,9 +1060,9 @@ class TestExperimentList(object):
     DETAILS_STDOUT = """+---------------+---------------+---------+
 | Name          | ID            | Status  |
 +---------------+---------------+---------+
-| dsfads        | ea2lfbbpdyzsq | created |
-| dsfads        | em6btk2vtb7it | created |
-| multinode_mpi | ew69ls0vy3eto | created |
+| dsfads        | ea2lfbbpdyzsq | pending |
+| dsfads        | em6btk2vtb7it | pending |
+| multinode_mpi | ew69ls0vy3eto | pending |
 +---------------+---------------+---------+
 
 Do you want to continue? [y/N]: