Support experiment action

Author: arjunsuresh

mlc/action.py

@@ -261,7 +261,7 @@ def add(self, i):
 
         target_name = i.get('target_name', self.action_type)
         target_path = os.path.join(repo_path, target_name)
-        if target_name == "cache":
+        if target_name in ["cache", "experiment"]:
             folder_name = f"""{i["script_alias"]}_{item_name or item_id[:8]}""" if i.get("script_alias") else item_name or item_id
         else:
             folder_name = item_name or item_id
@@ -424,7 +424,7 @@ def update(self, i):
             dict: Return code and message.
         """
         # Step 1: Search for items based on input tags
-        target_name = i.get('target_name',"cache")
+        target_name = i.get('target_name', i.get('target', "cache"))
         i['target_name'] = target_name
         ii = i.copy()
 
@@ -712,7 +712,7 @@ def search(self, i):
                 if target == "script":
                     non_variation_tags = [t for t in tags_split if not t.startswith("_")]
                     tags_to_match = non_variation_tags
-                elif target =="cache":
+                elif target in ["cache", "experiment"]:
                     tags_to_match = tags_split
                 else:
                     return {'return': 1, 'error': f"""Target {target} not handled in mlc yet"""}

mlc/action_factory.py

@@ -1,6 +1,7 @@
 from .repo_action import RepoAction
 from .script_action import ScriptAction
 from .cache_action import CacheAction
+from .experiment_action import ExperimentAction
 
 
 # Factory to get the appropriate action class
@@ -11,5 +12,6 @@ def get_action(target, parent):
 actions = {
         'repo': RepoAction,
         'script': ScriptAction,
-        'cache': CacheAction
+        'cache': CacheAction,
+        'experiment': ExperimentAction
     }

mlc/experiment_action.py

@@ -5,8 +5,6 @@
 
 class ExperimentAction(Action):
     def __init__(self, parent=None):
-        if parent is None:
-            parent = default_parent
         #super().__init__(parent)
         self.parent = parent
         self.__dict__.update(vars(parent))