more user-friendly flags for "cm init"

mlcommons · Aug 12, 2024 · 2f62c09 · 2f62c09
1 parent 532850b
commit 2f62c09
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/cm/cmind/repo/automation/core/module.py b/cm/cmind/repo/automation/core/module.py
@@ -211,8 +211,8 @@ def init(self, i):
           (CM input dict):
 
           (quiet) (bool): if True, skip asking questions about sudo, etc
-          (repo_name) (str): automation repository to pull ('mlcommons@cm4mlops' by default) 
-          (repo_branch) (str): branch to use ('' by default)
+          (repo) (str): automation repository to pull ('mlcommons@cm4mlops' by default) 
+          (branch) (str): branch to use ('' by default)
 
 
         Returns:
@@ -226,10 +226,10 @@ def init(self, i):
 
         quiet = i.get('quiet', False)
 
-        repo_name = i.get('repo_name', '')
+        repo_name = i.get('repo', '')
         if repo_name == '': repo_name = 'mlcommons@cm4mlops'
 
-        repo_branch = i.get('repo_branch', '')
+        repo_branch = i.get('branch', '')
 
         r = cm_init.run(quiet = quiet, repo_name = repo_name, repo_branch = repo_branch)
         if r['return']>0: return r