ソースを参照

[multi-python] remove the pytorch and unnecessary libs from req file

Change-Id: I52ade9334b6a861d5b9bdfe89ce8cc2e5258d06b
agl29 6 ヶ月 前
コミット
16be4adc7b
1 ファイル変更1 行追加9 行削除
  1. 1 9
      desktop/core/generate_requirements.py

+ 1 - 9
desktop/core/generate_requirements.py

@@ -47,12 +47,10 @@ class RequirementsGenerator:
             "django-celery-results==2.5.1",
             "django-cors-headers==3.13.0",
             "django-crequest==2018.5.11",
-            "django-debug-panel==0.8.3",
-            "django-debug-toolbar==3.6.0",
             "django-extensions==3.2.1",
             "django-ipware==3.0.2",
             "django_opentracing==1.1.0",
-            "django_prometheus==1.0.15",
+            "django_prometheus==2.3.1",
             "django-webpack-loader==1.0.0",
             "djangomako==1.3.2",
             "djangorestframework-simplejwt==5.2.1",
@@ -176,11 +174,6 @@ class RequirementsGenerator:
                 "Markdown==3.8"
             ],
         }
-        self.pytorch_requirements = {
-            "default": [
-                "--extra-index-url \"https://cloudera-build-us-west-1.vpc.cloudera.com/whl/cpu\" torch==2.2.2+cpu torchvision==0.17.2+cpu",
-            ]
-        }
         self.arch_requirements_map = {
             "ppc64le": self.ppc64le_requirements,
             "x86_64": self.x86_64_requirements,
@@ -202,7 +195,6 @@ class RequirementsGenerator:
         arch_reqs = self.arch_requirements_map[self.arch]
         self.requirements.extend(arch_reqs.get(self.python_version_string, arch_reqs["default"]))
         self.requirements.extend(self.copy_local_requirements(self.python_version_string))
-        self.requirements.extend(self.pytorch_requirements.get(self.python_version_string, self.pytorch_requirements["default"]))
 
         with open(f"{this_dir}/requirements-{self.arch}-{self.python_version_string}.txt", "w") as f:
             f.write("\n".join(self.requirements))