Fix access function

mlcommons · Jan 23, 2025 · 71f54fb · 71f54fb
1 parent 19c420b
commit 71f54fb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mlc/main.py b/mlc/main.py
@@ -1186,8 +1186,8 @@ def get_action(target):
 
 def access(i):
     action = i['action']
-    automation = i['automation']
-    action_class = get_action(automation)
+    target = i.get('target', i['automation'])
+    action_class = get_action(target)
     r = action_class.access(i)
     return r