Skip to content

Commit 388e576

Browse files
fix: update dpdata dependency to point to the development branch (#329)
* fix: update dpdata dependency to point to the development branch * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 6747a03 commit 388e576

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

lambench/workflow/dflow.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
from dflow.plugins.bohrium import BohriumDatasetsArtifact, create_job_group
1313
from dflow.plugins.dispatcher import DispatcherExecutor
1414
from dflow.python import OP, Artifact, PythonOPTemplate
15+
import dpdata
1516

1617
import lambench
1718
from lambench.models.basemodel import BaseLargeAtomModel
@@ -58,7 +59,9 @@ def submit_tasks_dflow(
5859
run_task_op, # type: ignore
5960
image=model.virtualenv,
6061
envs={k: v for k, v in os.environ.items() if k.startswith("MYSQL")},
61-
python_packages=[Path(package.__path__[0]) for package in [lambench]],
62+
python_packages=[
63+
Path(package.__path__[0]) for package in [lambench, dpdata]
64+
],
6265
),
6366
parameters={
6467
"task": task,

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ dependencies = [
1313
"pyyaml",
1414
"SQLAlchemy[pymysql]",
1515
"tqdm",
16-
"dpdata >= 0.2.22",
16+
"dpdata @ git+https://github.com/deepmodeling/dpdata.git@devel#egg=dpdata",
1717
"pandas",
1818
]
1919

0 commit comments

Comments
 (0)