Skip to content

Commit 0f60fce

Browse files
author
chibu
committed
moved packaged pinning from guidellm to base task
1 parent 1cf585b commit 0f60fce

File tree

2 files changed

+5
-6
lines changed

2 files changed

+5
-6
lines changed

src/automation/tasks/base_task.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,11 @@ def __init__(
2424
else:
2525
packages = base_packages
2626

27+
# keep only the pinned version of a library
28+
for pkg in packages:
29+
if "==" in pkg and pkg.split("==")[0] in packages:
30+
lib_name = pkg.split("==")[0]
31+
packages.remove(lib_name)
2732
print(packages)
2833

2934
self.project_name = project_name

src/automation/tasks/guidellm.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -46,12 +46,6 @@ def __init__(
4646
if "packages" in config_kwargs:
4747
packages = list(set(packages + config_kwargs.pop("packages")))
4848

49-
# keep only the pinned version of a library
50-
for pkg in packages:
51-
if "==" in pkg and pkg.split("==")[0] in packages:
52-
lib_name = pkg.split("==")[0]
53-
packages.remove(lib_name)
54-
5549
# Initialize base parameters
5650
super().__init__(
5751
project_name=project_name,

0 commit comments

Comments
 (0)