Make core deps server deps (#1882)

chaiNNer-org · Jun 20, 2023 · 699f43e · 699f43e
1 parent 9d75c49
commit 699f43e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 29 deletions.
diff --git a/backend/src/dependencies/install_core_deps.py b/backend/src/dependencies/install_core_deps.py
diff --git a/backend/src/dependencies/install_server_deps.py b/backend/src/dependencies/install_server_deps.py
@@ -39,6 +39,16 @@
         "display_name": "Sanic-Cors",
         "version": "2.2.0",
     },
+    {
+        "package_name": "pynvml",
+        "display_name": "pynvml",
+        "version": "11.5.0",
+    },
+    {
+        "package_name": "typing_extensions",
+        "display_name": "typing_extensions",
+        "version": "4.6.2",
+    },
 ]
 
 install_dependencies_sync(deps)
diff --git a/backend/src/server.py b/backend/src/server.py
@@ -516,21 +516,12 @@ async def update_progress(message: str, percent: float):
         timeout=1,
     )
 
-    await update_progress("Installing dependencies...", 0.0)
-
-    # Now we can install the other dependencies
-    importlib.import_module("dependencies.install_core_deps")
-
-    await update_progress("Importing nodes...", 0.5)
+    await update_progress("Importing nodes...", 0.0)
 
     logger.info("Importing nodes...")
 
-    # Update progress between 0.5 and 1.0
-    async def update_sub_progress(message: str, percent: float):
-        await update_progress(message, 0.5 + percent / 2)
-
     # Now we can load all the nodes
-    await import_packages(AppContext.get(sanic_app).config, update_sub_progress)
+    await import_packages(AppContext.get(sanic_app).config, update_progress)
 
     logger.info("Sending backend ready...")