Merge pull request #25 from dule1322/CTX-4251

CTX-4251: Update "model-queue/custom" to "run" endpoint
coretex-ai · Jul 11, 2023 · 9415689 · 9415689
2 parents a56f872 + e361f67
commit 9415689
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 14 deletions.
diff --git a/coretex/coretex/experiment/experiment.py b/coretex/coretex/experiment/experiment.py
@@ -390,7 +390,7 @@ def createQiimeArtifact(self, rootArtifactFolderName: str, qiimeArtifactPath: Pa
         #         logging.getLogger("coretexpylib").warning(f">> [Coretex] Failed to upload {localFilePath} to {remoteFilePath}")
 
     @classmethod
-    def startCustomExperiment(
+    def run(
         cls,
         projectId: int,
         nodeId: Union[int, str],
@@ -438,7 +438,7 @@ def startCustomExperiment(
                 ]
             \b
             >>> try:
-                    experiment = Experiment.startCustomExperiment(
+                    experiment = Experiment.run(
                         projectId = 1023,
                         nodeId = 23,
                         name = "Dummy Custom Experiment
@@ -457,17 +457,13 @@ def startCustomExperiment(
         if parameters is None:
             parameters = []
 
-        response = networkManager.genericJSONRequest(
-            f"{cls._endpoint()}/custom",
-            RequestType.post,
-            parameters={
-                "sub_project_id": projectId,
-                "service_id": nodeId,
-                "name": name,
-                "description": description,
-                "parameters": parameters
-            }
-        )
+        response = networkManager.genericJSONRequest("run", RequestType.post, {
+            "sub_project_id": projectId,
+            "service_id": nodeId,
+            "name": name,
+            "description": description,
+            "parameters": parameters
+        })
 
         if response.hasFailed():
             raise NetworkRequestError(response, "Failed to create experiment")

diff --git a/coretex/project/local.py b/coretex/project/local.py
@@ -113,7 +113,7 @@ def processLocal(args: Optional[List[str]] = None) -> Tuple[int, ProjectCallback
 
     parameters = ExperimentParameter.readExperimentConfig()
 
-    experiment: Experiment = Experiment.startCustomExperiment(
+    experiment: Experiment = Experiment.run(
         parser.projectId,
         # Dummy Local node ID, hardcoded as it is only a temporary solution,
         # backend will add a new ExperimentType (local) which does not require a specific