From d428fddb1ffd4acb5223616f32566ede1f67a81f Mon Sep 17 00:00:00 2001 From: Edgard Marx Date: Tue, 9 Apr 2024 16:05:06 +0200 Subject: [PATCH] issue lowercase --- imr/imr_cli.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/imr/imr_cli.py b/imr/imr_cli.py index 93640f9..c60e854 100644 --- a/imr/imr_cli.py +++ b/imr/imr_cli.py @@ -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() @@ -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() @@ -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__":