Skip to content

Commit

Permalink
Fix cli multiple option
Browse files Browse the repository at this point in the history
  • Loading branch information
FNTwin committed Feb 29, 2024
1 parent 984e971 commit ef8264a
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 14 deletions.
16 changes: 2 additions & 14 deletions openqdc/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,12 +21,7 @@ def exist_dataset(dataset):

@app.command()
def download(
datasets: Annotated[
List[str],
typer.Option(
help="List of datasets to download",
),
],
datasets: List[str],
overwrite: Annotated[
bool,
typer.Option(
Expand Down Expand Up @@ -66,14 +61,7 @@ def datasets():


@app.command()
def fetch(
datasets: Annotated[
List[str],
typer.Option(
help="List of datasets to fetch",
),
],
):
def fetch(datasets: List[str]):
"""
Download the raw datasets files from openQDC.
"""
Expand Down
1 change: 1 addition & 0 deletions openqdc/raws/fetch.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""Script to download the molecule3d dataset from Google Drive."""

import gzip
import os
import shutil
Expand Down
1 change: 1 addition & 0 deletions openqdc/utils/io.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""IO utilities for mlip package"""

import json
import os
import pickle as pkl
Expand Down

0 comments on commit ef8264a

Please sign in to comment.