diff --git a/mlc/index.py b/mlc/index.py index b1db888b9..2e6abbeff 100644 --- a/mlc/index.py +++ b/mlc/index.py @@ -48,6 +48,7 @@ def add(self, meta, folder_type, path, repo): "path": path, "repo": repo }) + self._save_indices() def get_index(self, folder_type, uid): for index in range(len(self.indices[folder_type])): @@ -71,6 +72,7 @@ def update(self, meta, folder_type, path, repo): "path": path, "repo": repo } + self._save_indices() def rm(self, meta, folder_type, path): uid = meta['uid'] @@ -79,6 +81,7 @@ def rm(self, meta, folder_type, path): logger.warning(f"Index is not having the {folder_type} item {path}") else: del(self.indices[folder_type][index]) + self._save_indices() def build_index(self): """ diff --git a/mlc/repo_action.py b/mlc/repo_action.py index 677fc2495..491fa334d 100644 --- a/mlc/repo_action.py +++ b/mlc/repo_action.py @@ -296,8 +296,8 @@ def pull(self, run_args): repo_url = run_args.get('repo', run_args.get('url', 'repo')) if not repo_url or repo_url == "repo": for repo_object in self.repos: - repo_folder_name = os.path.basename(repo_object.path) - if "@" in repo_folder_name: + if os.path.exists(os.path.join(repo_object.path, ".git")): + repo_folder_name = os.path.basename(repo_object.path) res = self.pull_repo(repo_folder_name) if res['return'] > 0: return res diff --git a/pyproject.toml b/pyproject.toml index 7930e1c7a..620d12cbc 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,8 @@ build-backend = "setuptools.build_meta" [project] name = "mlcflow" -version = "1.0.6" +version = "1.0.7" + description = "An automation interface for ML applications" authors = [ { name = "MLCommons", email = "systems@mlcommons.org" }