Skip to content

Commit 8d388e2

Browse files
committed
add install script
1 parent 356b933 commit 8d388e2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lambench/workflow/dflow.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,9 +61,9 @@ def submit_tasks_dflow(
6161
image=model.virtualenv,
6262
envs={k: v for k, v in os.environ.items() if k.startswith("MYSQL")},
6363
python_packages=[
64-
Path(package.__path__[0])
65-
for package in [lambench, dpdata]
64+
Path(package.__path__[0]) for package in [lambench, dpdata]
6665
],
66+
pre_script="pip install dftd3",
6767
),
6868
parameters={
6969
"task": task,

0 commit comments

Comments
 (0)