synapse-sdk 1.0.0a17__tar.gz → 1.0.0a19__tar.gz

{synapse_sdk-1.0.0a17/synapse_sdk.egg-info → synapse_sdk-1.0.0a19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: synapse-sdk
-Version: 1.0.0a17
+Version: 1.0.0a19
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT

{synapse_sdk-1.0.0a17 → synapse_sdk-1.0.0a19}/synapse_sdk/plugins/categories/base.py

@@ -147,7 +147,8 @@ class Action:
                 return self.start_by_restapi()
             else:
                 result = self.start()
-                self.post_action_by_job(result)
+                if self.job_id:
+                    self.post_action_by_job(result)
                 return result
         return getattr(self, f'start_by_{self.method.value}')()
 
@@ -182,8 +183,7 @@ class Action:
 
         kwargs['direct'] = True
         try:
-            if not ray.is_initialized():
-                ray.init(address=self.envs['RAY_ADDRESS'], ignore_reinit_error=True)
+            self.ray_init()
             return ray.get(run_task.remote(self.category.value, self.name, *args, **kwargs))
         except RayTaskError as e:
             raise ActionError(e.cause)
@@ -227,12 +227,17 @@ class Action:
             raise ActionError('Unable to connect to serve application')
 
     def post_action_by_job(self, result):
-        if self.client:
-            job_client = self.get_job_client()
-            logs = job_client.get_job_logs(self.job_id).split('\n')
-            self.client.update_job(self.job_id, data={'result': result or {}, 'console_logs': logs})
+        job_client = self.get_job_client()
+        logs = job_client.get_job_logs(self.job_id).split('\n')
+        self.client.update_job(self.job_id, data={'result': result or {}, 'console_logs': logs})
 
     def get_job_client(self):
         from ray.dashboard.modules.job.sdk import JobSubmissionClient
 
         return JobSubmissionClient(address=self.envs.get('RAY_DASHBOARD_URL'))
+
+    def ray_init(self):
+        import ray
+
+        if not ray.is_initialized():
+            ray.init(address=self.envs['RAY_ADDRESS'], ignore_reinit_error=True)

{synapse_sdk-1.0.0a17 → synapse_sdk-1.0.0a19}/synapse_sdk/plugins/categories/neural_net/actions/deployment.py

@@ -11,11 +11,17 @@ class DeploymentAction(Action):
     method = RunMethod.JOB
 
     def get_actor_options(self):
-        return {'runtime_env': self.get_runtime_env()}
+        options = {'runtime_env': self.get_runtime_env()}
+        for option in ['num_cpus', 'num_gpus']:
+            option_value = self.params.get(option)
+            if option_value:
+                options[option] = option_value
+        return options
 
     def start(self):
         from ray import serve
 
+        self.ray_init()
         deployment = serve.deployment(ray_actor_options=self.get_actor_options())(self.entrypoint)
         serve.delete(self.plugin_release.code)
         # TODO add run object
@@ -26,7 +32,7 @@ class DeploymentAction(Action):
         return {'serve_application': serve_application['id'] if serve_application else None}
 
     def create_serve_application(self):
-        if self.client:
+        if self.job_id:
             try:
                 serve_application = self.ray_client.get_serve_application(self.plugin_release.code)
                 return self.client.create_serve_application({

{synapse_sdk-1.0.0a17 → synapse_sdk-1.0.0a19/synapse_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: synapse-sdk
-Version: 1.0.0a17
+Version: 1.0.0a19
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT