diff --git a/script/clean-nvidia-mlperf-inference-scratch-space/customize.py b/script/clean-nvidia-mlperf-inference-scratch-space/customize.py index aeb40f6c0..6724c845e 100644 --- a/script/clean-nvidia-mlperf-inference-scratch-space/customize.py +++ b/script/clean-nvidia-mlperf-inference-scratch-space/customize.py @@ -30,7 +30,7 @@ def preprocess(i): cache_rm_tags = "nvidia-harness,_download_model,_sdxl" cache_rm_tags = cache_rm_tags + extra_cache_rm_tags - mlc = i['automation'].action_target + mlc = i['automation'].action_object if cache_rm_tags: r = mlc.access({'action': 'rm', 'automation': 'cache',