Merge pull request #95 from roboflow/deploy-follow

New Deployment Status Routes

Author: Jacobsolawetz

roboflow/core/project.py

@@ -93,6 +93,7 @@ def versions(self):
                 local=None,
                 workspace=self.__workspace,
                 project=self.__project_name,
+                public=self.public,
             )
             version_array.append(version_object)
         return version_array
@@ -251,6 +252,7 @@ def version(self, version_number, local=None):
                     local=local,
                     workspace=self.__workspace,
                     project=self.__project_name,
+                    public=self.public,
                 )
                 return vers
 

roboflow/core/version.py

@@ -41,6 +41,7 @@ def __init__(
         local,
         workspace,
         project,
+        public,
     ):
         if api_key in DEMO_KEYS:
             if api_key == "coco-128-sample":
@@ -72,6 +73,7 @@ def __init__(
             self.model_format = model_format
             self.workspace = workspace
             self.project = project
+            self.public = public
             if "exports" in version_dict.keys():
                 self.exports = version_dict["exports"]
             else:
@@ -372,12 +374,18 @@ def deploy(self, model_type: str, model_path: str) -> None:
         try:
             res.raise_for_status()
 
-            print(
-                f"View the status of your deployment at: {APP_URL}/{self.workspace}/{self.project}/{self.version}"
-            )
-            print(
-                f"Share your model with the world at: {UNIVERSE_URL}/{self.workspace}/{self.project}/{self.version}"
-            )
+            if self.public:
+                print(
+                    f"View the status of your deployment at: {APP_URL}/{self.workspace}/{self.project}/deploy/{self.version}"
+                )
+                print(
+                    f"Share your model with the world at: {UNIVERSE_URL}/{self.workspace}/{self.project}/model/{self.version}"
+                )
+            else:
+                print(
+                    f"View the status of your deployment at: {APP_URL}/{self.workspace}/{self.project}/deploy/{self.version}"
+                )
+
         except Exception as e:
             print(f"An error occured when uploading the model: {e}")
 

tests/helpers.py

@@ -5,6 +5,7 @@
 def get_version(
     api_key="test-api-key",
     project_name="Test Project Name",
+    public=True,
     version_number="1",
     type=TYPE_OBJECT_DETECTION,
     workspace_name="Test Workspace Name",
@@ -46,4 +47,5 @@ def get_version(
         local=None,
         workspace=workspace_name,
         project=project_name,
+        public=public
     )