possible fix for #31

ai-models · Dec 7, 2022 · b2bdba8 · b2bdba8
1 parent 4fe9875
commit b2bdba8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/cli/base_funcs.py b/cli/base_funcs.py
@@ -135,6 +135,9 @@ def get_model_path(name_version):
 
 # a function that updates aimodels-lock.json
 def update_ai_models_lock(name, version, path):
+    # make name and version case insensitive
+    name = name.lower()
+    version = version.lower()
     # get current working directory
     cwd = os.getcwd()
     aimodels_lock_file = os.path.join(cwd, "aimodels-lock.json")

diff --git a/cli/install.py b/cli/install.py
@@ -92,7 +92,7 @@ def install(name_version: Optional[str] = typer.Argument(None),
                         typer.echo(f"Error: Model {name}:{version} not found")
                         return
 
-                    save_path = os.path.join(aimmApp.main_dir, name, version)
+                    save_path = os.path.join(aimmApp.main_dir, name.lower(), version.lower())
                     if not os.path.exists(save_path):
                         os.makedirs(save_path)
                     # if auth_required is set, check for creds
@@ -132,6 +132,6 @@ def install(name_version: Optional[str] = typer.Argument(None),
                     json.dump(aimmApp.installed, file, indent=4)
                 typer.echo(f"Installed {name}:{version}!")
     else: 
-        save_path = os.path.join(aimmApp.main_dir, name, version)
+        save_path = os.path.join(aimmApp.main_dir, name.lower(), version.lower())
         base_funcs.update_ai_models_lock(name, version, save_path)
         typer.echo(f"Found Local: {name}:{version}")