Skip to content

Commit

Permalink
feat: support datasetbuilder --version
Browse files Browse the repository at this point in the history
Signed-off-by: Jinzhe Zeng <[email protected]>
  • Loading branch information
njzjz committed May 25, 2024
1 parent b257b95 commit bc20946
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions mddatasetbuilder/datasetbuilder.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@
read_compressed_block,
run_mp,
)
from ._version import version as __version__


class DatasetBuilder:
Expand Down Expand Up @@ -725,6 +726,11 @@ def _commandline():
type=float,
default=0.0,
)
parser.add_argument(

Check warning on line 729 in mddatasetbuilder/datasetbuilder.py

View check run for this annotation

Codecov / codecov/patch

mddatasetbuilder/datasetbuilder.py#L729

Added line #L729 was not covered by tests
"--version",
action="version",
version=f"MDDatasetBuilder {__version__}",
)
args = parser.parse_args()
DatasetBuilder(
atomname=args.atomname,
Expand Down

0 comments on commit bc20946

Please sign in to comment.