feat: Update nse.

AmbiqAI · Aug 21, 2024 · db16d66 · db16d66
1 parent 3d5f342
commit db16d66
Show file tree

Hide file tree

Showing 6 changed files with 41 additions and 14 deletions.
diff --git a/docs/api/backends/backend.md b/docs/api/backends/backend.md
@@ -14,4 +14,4 @@ for backend in hk.BackendFactory.list():
     print(f"Backend name: {backend} - {hk.BackendFactory.get(backend)}")
 ```
 
-::: neuralspot_edge.utils.ItemFactory
+<!-- ::: neuralspot_edge.utils.ItemFactory -->
diff --git a/docs/api/datasets/factory.md b/docs/api/datasets/factory.md
@@ -13,4 +13,4 @@ for dataset in hk.DatasetFactory.list():
     print(f"Dataset name: {dataset} - {hk.DatasetFactory.get(dataset)}")
 ```
 
-::: neuralspot_edge.utils.ItemFactory
+<!-- ::: neuralspot_edge.utils.ItemFactory -->
diff --git a/docs/api/tasks/factory.md b/docs/api/tasks/factory.md
@@ -11,4 +11,4 @@ for model in hk.TaskFactory.list():
     print(f"Task name: {model} - {hk.TaskFactory.get(model)}")
 ```
 
-::: neuralspot_edge.utils.ItemFactory
+<!-- ::: neuralspot_edge.utils.ItemFactory -->
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -58,6 +58,7 @@ orjson = "^3.9.13"
 physiokit = "^0.8.1"
 requests = "^2.31.0"
 argdantic = {extras = ["all"], version = "^1.0.0"}
+neuralspot-edge = "^0.1.5"
 
 [tool.poetry.group.dev.dependencies]
 ipython = "^8.21.0"

diff --git a/scripts/upload_to_model_zoo.py b/scripts/upload_to_model_zoo.py
@@ -78,18 +78,19 @@ def upload_to_model_zoo(
     parser()
 
 """
-python ./scripts/upload_to_model_zoo.py -t rhythm -s ./results/arr-2-eff-sm -n arr-2-eff-sm -v latest
-python ./scripts/upload_to_model_zoo.py -t rhyhtm -s ./results/arr-4-eff-sm -n arr-4-eff-sm -v latest
+python ./scripts/upload_to_model_zoo.py -t rhythm -s ./results/arr-2-eff-sm -n arr-2-eff-sm -v v1.0
+python ./scripts/upload_to_model_zoo.py -t rhyhtm -s ./results/arr-4-eff-sm -n arr-4-eff-sm -v v1.0
 
-python ./scripts/upload_to_model_zoo.py -t beat -s ./results/beat-2-eff-sm -n beat-2-eff-sm -v latest
-python ./scripts/upload_to_model_zoo.py -t beat -s ./results/beat-3-eff-sm -n beat-3-eff-sm -v latest
+python ./scripts/upload_to_model_zoo.py -t beat -s ./results/beat-2-eff-sm -n beat-2-eff-sm -v v1.0
+python ./scripts/upload_to_model_zoo.py -t beat -s ./results/beat-3-eff-sm -n beat-3-eff-sm -v v1.0
 
-python ./scripts/upload_to_model_zoo.py -t denoise -s ./results/den-ppg-tcn-sm -n den-ppg-tcn-sm -v latest
-python ./scripts/upload_to_model_zoo.py -t denoise -s ./results/den-tcn-sm -n den-tcn-sm -v latest
+python ./scripts/upload_to_model_zoo.py -t denoise -s ./results/den-ppg-tcn-sm -n den-ppg-tcn-sm -v v1.0
+python ./scripts/upload_to_model_zoo.py -t denoise -s ./results/den-tcn-sm -n den-tcn-sm -v v1.0
 
-python ./scripts/upload_to_model_zoo.py -t foundation -s ./results/fnd-eff-sm -n fnd-eff-sm -v latest
+python ./scripts/upload_to_model_zoo.py -t foundation -s ./results/fnd-eff-sm -n fnd-eff-sm -v v1.0
 
-python ./scripts/upload_to_model_zoo.py -t segmentation -s ./results/seg-2-tcn-sm -n seg-2-tcn-sm -v latest
-python ./scripts/upload_to_model_zoo.py -t segmentation -s ./results/seg-4-tcn-sm -n seg-4-tcn-sm -v latest
-python ./scripts/upload_to_model_zoo.py -t segmentation -s ./results/seg-ppg-2-tcn-sm -n seg-ppg-2-tcn-sm -v latest
+python ./scripts/upload_to_model_zoo.py -t segmentation -s ./results/seg-2-tcn-sm -n seg-2-tcn-sm -v v1.0
+python ./scripts/upload_to_model_zoo.py -t segmentation -s ./results/seg-4-tcn-sm -n seg-4-tcn-sm -v v1.0
+python ./scripts/upload_to_model_zoo.py -t segmentation -s ./results/seg-4-tcn-lg -n seg-4-tcn-lg -v v1.0
+python ./scripts/upload_to_model_zoo.py -t segmentation -s ./results/seg-ppg-2-tcn-sm -n seg-ppg-2-tcn-sm -v v1.0
 """