Support remote-run for mlc, fix for logging (#190)

Author: arjunsuresh

mlc/main.py

@@ -144,7 +144,7 @@ def build_parser(pre_args):
         p.add_argument('extra', nargs=argparse.REMAINDER)
 
     # Script-only
-    for action in ['docker', 'docker-run', 'experiment', 'doc', 'lint']:
+    for action in ['docker', 'docker-run', 'experiment', 'remote-run', 'doc', 'lint']:
         p = subparsers.add_parser(action, add_help=False)
         p.add_argument('target', choices=['script', 'run'])
         p.add_argument('details', nargs='?', help='Details or identifier (optional)')
@@ -161,7 +161,7 @@ def configure_logging(args):
         args.extra[:] = [log_flag_aliases.get(a, a) for a in args.extra]
         for flag, level in log_levels.items():
             if flag in args.extra:
-                logging.getLogger().setLevel(level)
+                logger.setLevel(level)
                 args.extra.remove(flag)
 
 
@@ -176,7 +176,7 @@ def build_run_args(args):
     if args.command in ['pull', 'rm', 'add', 'find'] and args.target == "repo":
         run_args['repo'] = args.details
 
-    if args.command in ['docker', 'docker-run', 'experiment', 'doc', 'lint'] and args.target == "run":
+    if args.command in ['docker', 'docker-run', 'experiment', 'remote-run', 'doc', 'lint'] and args.target == "run":
         run_args['target'] = 'script'
         args.target = "script"
 

mlc/script_action.py

@@ -235,6 +235,8 @@ def call_script_module_function(self, function_name, run_args):
                 result = automation_instance.test(run_args)  # Pass args to the run method
             elif function_name == "experiment":
                 result = automation_instance.experiment(run_args)  # Pass args to the experiment method
+            elif function_name == "remote_run":
+                result = automation_instance.remote_run(run_args)  # Pass args to the experiment method
             elif function_name == "help":
                 result = automation_instance.help(run_args)  # Pass args to the help method
             elif function_name == "doc":

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mlcflow"
-version = "1.1.1"
+version = "1.1.6"
 
 
 
@@ -47,6 +47,7 @@ py-modules = []
 [project.scripts]
 mlc = "mlc.main:main"
 mlcr = "mlc.main:mlcr"
+mlcrr = "mlc.main:mlcrr"
 mlcd = "mlc.main:mlcd"
 mlce = "mlc.main:mlce"
 mlct = "mlc.main:mlct"