fix: add mxnet-alas for linux (#3938)

LmeSzinc · Jun 28, 2024 · bb16429 · bb16429
1 parent 79edef0
commit bb16429
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 2 deletions.
diff --git a/deploy/docker/requirements.txt b/deploy/docker/requirements.txt
@@ -40,4 +40,5 @@ pywebio==1.6.2
 starlette==0.14.2
 uvicorn[standard]==0.17.6
 zerorpc==0.6.3
-pyzmq==22.3.0
+pyzmq==22.3.0
+mxnet-alas==0.0.5
diff --git a/deploy/docker/requirements_generator.py b/deploy/docker/requirements_generator.py
@@ -41,6 +41,9 @@ def docker_requirements_generate(requirements_in='requirements-in.txt'):
 
     logger.info(f'Generate requirements for Docker image')
     lock = {}
+    expand = {
+        'mxnet-alas': '0.0.5'
+    }
     new = {}
     logger.info(requirements)
     for name, version in requirements.items():
@@ -54,6 +57,8 @@ def docker_requirements_generate(requirements_in='requirements-in.txt'):
             version = lock[name] if not isinstance(lock[name], dict) else lock[name]['version']
             name = name if not isinstance(lock[name], dict) else lock[name]['name']
         new[name] = version
+    for name, version in expand.items():
+        new[name] = version
 
     write_file(os.path.join(BASE_FOLDER, f'./requirements.txt'), data=new)
 

diff --git a/deploy/headless/requirements.txt b/deploy/headless/requirements.txt
@@ -40,4 +40,5 @@ pywebio==1.6.2
 starlette==0.14.2
 uvicorn[standard]==0.17.6
 zerorpc==0.6.3
-pyzmq==22.3.0
+pyzmq==22.3.0
+mxnet-alas==0.0.5
diff --git a/deploy/headless/requirements_generator.py b/deploy/headless/requirements_generator.py
@@ -57,6 +57,9 @@ def headless_requirements_generate(requirements_in='requirements-in.txt'):
         'tqdm': '4.65.0',
         'wrapt': '1.15.0'
     }
+    expand = {
+        'mxnet-alas': '0.0.5'
+    }
     new = {}
     logger.info(requirements)
     for name, version in requirements.items():
@@ -66,6 +69,8 @@ def headless_requirements_generate(requirements_in='requirements-in.txt'):
             version = lock[name] if not isinstance(lock[name], dict) else lock[name]['version']
             name = name if not isinstance(lock[name], dict) else lock[name]['name']
         new[name] = version
+    for name, version in expand.items():
+        new[name] = version
 
     write_file(os.path.join(BASE_FOLDER, f'./requirements.txt'), data=new)