Fix cache_expiration, support mlc prune cache (#213)

Author: amd-arsuresh

VERSION

@@ -1 +1 @@
-1.1.16
+1.1.17

mlc/cache_action.py

@@ -1,6 +1,7 @@
 from .action import Action
 import os
 import json
+import time
 from . import utils
 from .logger import logger
 
@@ -59,7 +60,7 @@ def search(self, i):
                 continue  # skip item
             '''
 
-            expiration_time = item_meta.get('expiration_time')
+            expiration_time = item_meta.get('cache_expiration')
             if expiration_time is not None and expiration_time < time.time():
                 continue  # skip expired item
 
@@ -158,6 +159,42 @@ def show(self, run_args):
 
         return {'return': 0}
 
+    
+
+    def prune(self, args):
+        """
+    ####################################################################################################################
+    Target: Cache
+    Action: Prune
+    ####################################################################################################################
+
+    Prune all expired cached items
+
+    Example Command:
+
+    mlc prune cache
+
+        """
+        self.action_type = "cache"
+        run_args = {"fetch_all": True}  # to fetch the details of all the caches generated
+        
+        res = self.search(run_args)
+        if res['return'] > 0:
+            return res
+        
+        for item in res['list']:
+            expiration_time = item.meta.get('cache_expiration')
+            if expiration_time is not None and expiration_time < time.time():
+                ii = {}
+                ii['f'] = True
+                ii['item'] = item.meta.get('uid')
+                if ii['item']:
+                    self.rm(ii)
+
+        return {'return': 0}
+        
+
+
     def list(self, args):
         """
     ####################################################################################################################

mlc/main.py

@@ -155,7 +155,7 @@ def build_parser(pre_args):
     subparsers = parser.add_subparsers(dest="command", required=not pre_args.help)
 
     # General commands
-    for action in ['run', 'pull', 'test', 'add', 'show', 'list', 'find', 'search', 'rm', 'cp', 'mv', 'help']:
+    for action in ['run', 'pull', 'test', 'add', 'show', 'list', 'find', 'search', 'rm', 'cp', 'mv', 'help', 'prune']:
         p = subparsers.add_parser(action, add_help=False)
         p.add_argument('target', choices=['repo', 'repos', 'script', 'cache'])
         p.add_argument('details', nargs='?', help='Details or identifier (optional)')

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mlcflow"
-version = "1.1.16"
+version = "1.1.17"
 
 description = "An automation interface tailored for CPU/GPU benchmarking"
 authors = [