Skip to content

Commit

Permalink
feat: support datasetbuilder --version (#219)
Browse files Browse the repository at this point in the history
Signed-off-by: Jinzhe Zeng <[email protected]>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
  • Loading branch information
njzjz and pre-commit-ci[bot] authored May 25, 2024
1 parent b257b95 commit 132cca8
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 @@ -31,6 +31,7 @@
from sklearn.cluster import MiniBatchKMeans

from ._logger import logger
from ._version import version as __version__
from .detect import Detect, DetectDump
from .utils import (
bytestolist,
Expand Down Expand Up @@ -725,6 +726,11 @@ def _commandline():
type=float,
default=0.0,
)
parser.add_argument(
"--version",
action="version",
version=f"MDDatasetBuilder {__version__}",
)
args = parser.parse_args()
DatasetBuilder(
atomname=args.atomname,
Expand Down

0 comments on commit 132cca8

Please sign in to comment.