Skip to content

Commit

Permalink
issue lowercase
Browse files Browse the repository at this point in the history
  • Loading branch information
Edgard Marx committed Apr 9, 2024
1 parent a8ed489 commit d428fdd
Showing 1 changed file with 10 additions and 10 deletions.
20 changes: 10 additions & 10 deletions imr/imr_cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,18 +13,18 @@
imr_dir = home + "/.imr"
imr_local: IMRLocal = IMRLocal(imr_dir)
imr_remote: IMRRemote = None
imr_config = None


def loadParams() -> None:
def load_params() -> None:
"""Load the default parameters from conf.yaml file."""
with open(imr_dir + "/config.yaml") as stream:
imrConfig = yaml.safe_load(stream)
with Path.open(imr_dir + "/config.yaml") as stream:
imr_config = yaml.safe_load(stream)


@click.group()
def cli() -> None:
"""Get the cli command options."""
# loadParams()
# add loadParams() later


@cli.group()
Expand Down Expand Up @@ -78,14 +78,14 @@ def rm(package: str, version: str) -> None:


@remote.command()
@click.argument("model_directory")
@click.argument("model_dir")
@click.argument("package")
@click.option(
"-v", "--version", type=str, default="latest", help="version of the model.", show_default=True
)
def push(model_directory: str, package: str, version: str) -> None:
def push(model_dir: str, package: str, version: str) -> None:
"""Push model to remote repository."""
imr_remote.push(model_directory, package, version)
imr_remote.push(model_dir, package, version)


@remote.command()
Expand All @@ -101,9 +101,9 @@ def push(model_directory: str, package: str, version: str) -> None:
@click.option(
"-v", "--version", type=str, default="latest", help="version of the model.", show_default=True
)
def pull(package: str, dir: str, version: str) -> None:
def pull(package: str, model_dir: str, version: str) -> None:
"""Pull model from remote repository."""
imr_remote.pull(dir, package, version)
imr_remote.pull(model_dir, package, version)


if __name__ == "__main__":
Expand Down

0 comments on commit d428fdd

Please sign in to comment.