handle if env not in dependency

mlcommons · Nov 29, 2024 · 0cee8c8 · 0cee8c8
1 parent 8de2cf9
commit 0cee8c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/automation/script/module.py b/automation/script/module.py
@@ -3526,7 +3526,7 @@ def _run_deps(self, deps, clean_env_keys_deps, env, state, const, const_state, a
                     "update_tags_from_env_with_prefix", {})
                 for t in update_tags_from_env_with_prefix:
                     for key in update_tags_from_env_with_prefix[t]:
-                        if str(d.get('env').get(key, '')).strip() != '':
+                        if str(d.get('env', {}).get(key, '')).strip() != '':
                             d['tags'] += "," + t + str(d.get('env')[key])
                         elif str(env.get(key, '')).strip() != '':
                             d['tags'] += "," + t + str(env[key])