Skip to content

Commit 16be4ad

Browse files
committed
[multi-python] remove the pytorch and unnecessary libs from req file
Change-Id: I52ade9334b6a861d5b9bdfe89ce8cc2e5258d06b
1 parent 533a5a6 commit 16be4ad

File tree

1 file changed

+1
-9
lines changed

1 file changed

+1
-9
lines changed

desktop/core/generate_requirements.py

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -47,12 +47,10 @@ def __init__(self):
4747
"django-celery-results==2.5.1",
4848
"django-cors-headers==3.13.0",
4949
"django-crequest==2018.5.11",
50-
"django-debug-panel==0.8.3",
51-
"django-debug-toolbar==3.6.0",
5250
"django-extensions==3.2.1",
5351
"django-ipware==3.0.2",
5452
"django_opentracing==1.1.0",
55-
"django_prometheus==1.0.15",
53+
"django_prometheus==2.3.1",
5654
"django-webpack-loader==1.0.0",
5755
"djangomako==1.3.2",
5856
"djangorestframework-simplejwt==5.2.1",
@@ -176,11 +174,6 @@ def __init__(self):
176174
"Markdown==3.8"
177175
],
178176
}
179-
self.pytorch_requirements = {
180-
"default": [
181-
"--extra-index-url \"https://cloudera-build-us-west-1.vpc.cloudera.com/whl/cpu\" torch==2.2.2+cpu torchvision==0.17.2+cpu",
182-
]
183-
}
184177
self.arch_requirements_map = {
185178
"ppc64le": self.ppc64le_requirements,
186179
"x86_64": self.x86_64_requirements,
@@ -202,7 +195,6 @@ def generate_requirements(self):
202195
arch_reqs = self.arch_requirements_map[self.arch]
203196
self.requirements.extend(arch_reqs.get(self.python_version_string, arch_reqs["default"]))
204197
self.requirements.extend(self.copy_local_requirements(self.python_version_string))
205-
self.requirements.extend(self.pytorch_requirements.get(self.python_version_string, self.pytorch_requirements["default"]))
206198

207199
with open(f"{this_dir}/requirements-{self.arch}-{self.python_version_string}.txt", "w") as f:
208200
f.write("\n".join(self.requirements))

0 commit comments

Comments
 (0)