synapse-sdk 1.0.0a11__py3-none-any.whl → 1.0.0a12__py3-none-any.whl

synapse_sdk/plugins/categories/neural_net/templates/config.yaml

@@ -1,9 +1,15 @@
+task: object_detection
+data_type: image
 actions:
   train:
     dataset: datamaker
     entrypoint: plugin.train.train
+    metrics:
+      iteration:
+        - loss
+        - miou
   deployment:
-    entrypoint: plugin.inference.InferenceDeployment
+    entrypoint: plugin.inference.MockNetInference
   inference:
     method: restapi
     endpoints:

synapse_sdk/plugins/categories/neural_net/templates/plugin/inference.py

@@ -1,4 +1,14 @@
-class InferenceDeployment:
-    def __call__(self):
-        message = 'hello datamaker'
-        return {'message': message}
+from ray import serve
+
+
+class MockNetInference:
+    model_id = None
+
+    @serve.multiplexed()
+    async def get_model(self, model_id: str):
+        return model_id
+
+    async def __call__(self):
+        model_id = await self.get_model(serve.get_multiplexed_model_id())
+        message = 'hello datamaker!!!'
+        return {'model_id': model_id, 'message': message}

synapse_sdk/plugins/categories/neural_net/templates/plugin/train.py

@@ -3,12 +3,11 @@ import time
 
 def train(run, dataset, hyperparameter, checkpoint=None):
     count_iterations = hyperparameter['iterations']
-
+    run.set_progress(0, count_iterations, category='train')
     for i in range(1, count_iterations + 1):
         time.sleep(0.5)
         loss = float(round((count_iterations - i) / count_iterations, 2))
         miou = 1 - loss
         run.log_metric('iteration', i, loss=loss, miou=miou)
         run.set_progress(i, count_iterations, category='train')
-
-    return {'weight': '/tmp/agent/test/a.txt', 'config': '/tmp/agent/test/b.txt'}
+    return './'

{synapse_sdk-1.0.0a11.dist-info → synapse_sdk-1.0.0a12.dist-info}/METADATA

@@ -1,11 +1,11 @@
 Metadata-Version: 2.1
 Name: synapse-sdk
-Version: 1.0.0a11
+Version: 1.0.0a12
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT
 Classifier: Programming Language :: Python :: 3
-Requires-Python: >=3.11
+Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: boto3

{synapse_sdk-1.0.0a11.dist-info → synapse_sdk-1.0.0a12.dist-info}/RECORD

@@ -52,11 +52,11 @@ synapse_sdk/plugins/categories/neural_net/actions/deployment.py,sha256=ASRY0-ijd
 synapse_sdk/plugins/categories/neural_net/actions/inference.py,sha256=i18bDDjkuF9V8nxxW-T_umNIOD-Jnq_MMjIjZc6W8n8,645
 synapse_sdk/plugins/categories/neural_net/actions/test.py,sha256=JY25eg-Fo6WbgtMkGoo_qNqoaZkp3AQNEypJmeGzEog,320
 synapse_sdk/plugins/categories/neural_net/actions/train.py,sha256=TPtCAfovmFyzCYRbiRtlrov9AjE6XPQLDB94W0M1OuU,4558
-synapse_sdk/plugins/categories/neural_net/templates/config.yaml,sha256=pvJFh_NHGZHHFfqrvZm6fHO5lsS7CudaIGboP6wbtY8,252
+synapse_sdk/plugins/categories/neural_net/templates/config.yaml,sha256=HHiXhCc8BEitfcCOy2Jjstc5IfAsl4yitTva8qYbk54,349
 synapse_sdk/plugins/categories/neural_net/templates/plugin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-synapse_sdk/plugins/categories/neural_net/templates/plugin/inference.py,sha256=glYYdRybG3vRe48bYIxvue7bCFGPlEZmYWRoyBsLdZw,123
+synapse_sdk/plugins/categories/neural_net/templates/plugin/inference.py,sha256=InfqKWJYi6sqiUnfPKHC5KYGhxckDaWZNQ202u-uVP4,366
 synapse_sdk/plugins/categories/neural_net/templates/plugin/test.py,sha256=kYyk7l4UtcDUAH4nkdVUGrHHHjxI4p1U13HSLnmGPyE,53
-synapse_sdk/plugins/categories/neural_net/templates/plugin/train.py,sha256=-QzWrfJGOk6EagQZmC1MMmiOEIr3X1sVGszMJjv5dFg,500
+synapse_sdk/plugins/categories/neural_net/templates/plugin/train.py,sha256=easC8kVGB6AfvUVFyyywTsatSIMaR0i7n3QqQNKObko,492
 synapse_sdk/plugins/categories/post_annotation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synapse_sdk/plugins/categories/post_annotation/actions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synapse_sdk/plugins/categories/post_annotation/actions/post_annotation.py,sha256=dkPqOPPMUoxn7h1e-momdhqezhTJ7wLf-Dyx04zKfjw,347
@@ -103,9 +103,9 @@ synapse_sdk/utils/pydantic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
 synapse_sdk/utils/pydantic/config.py,sha256=1vYOcUI35GslfD1rrqhFkNXXJOXt4IDqOPSx9VWGfNE,123
 synapse_sdk/utils/pydantic/errors.py,sha256=0v0T12eQBr1KrFiEOBu6KMaPK4aPEGEC6etPJGoR5b4,1061
 synapse_sdk/utils/pydantic/validators.py,sha256=G47P8ObPhsePmd_QZDK8EdPnik2CbaYzr_N4Z6En8dc,193
-synapse_sdk-1.0.0a11.dist-info/LICENSE,sha256=bKzmC5YAg4V1Fhl8OO_tqY8j62hgdncAkN7VrdjmrGk,1101
-synapse_sdk-1.0.0a11.dist-info/METADATA,sha256=pPWxwlLgyGQBjBkz7eQ9zP-ptu28IS4cSoig73hzZZ4,1049
-synapse_sdk-1.0.0a11.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-synapse_sdk-1.0.0a11.dist-info/entry_points.txt,sha256=VNptJoGoNJI8yLXfBmhgUefMsmGI0m3-0YoMvrOgbxo,48
-synapse_sdk-1.0.0a11.dist-info/top_level.txt,sha256=ytgJMRK1slVOKUpgcw3LEyHHP7S34J6n_gJzdkcSsw8,12
-synapse_sdk-1.0.0a11.dist-info/RECORD,,
+synapse_sdk-1.0.0a12.dist-info/LICENSE,sha256=bKzmC5YAg4V1Fhl8OO_tqY8j62hgdncAkN7VrdjmrGk,1101
+synapse_sdk-1.0.0a12.dist-info/METADATA,sha256=cPJ9URHKdZwjsC4lFzzmMIx4bm-LET60fvKUMquaJC4,1049
+synapse_sdk-1.0.0a12.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+synapse_sdk-1.0.0a12.dist-info/entry_points.txt,sha256=VNptJoGoNJI8yLXfBmhgUefMsmGI0m3-0YoMvrOgbxo,48
+synapse_sdk-1.0.0a12.dist-info/top_level.txt,sha256=ytgJMRK1slVOKUpgcw3LEyHHP7S34J6n_gJzdkcSsw8,12
+synapse_sdk-1.0.0a12.dist-info/RECORD,,