use metaflow_config.

Author: RikishK

metaflow/plugins/aws/batch/batch_decorator.py

@@ -188,15 +188,13 @@ def __init__(self, attributes=None, statically_defined=False):
         if self.attributes["trainium"] is not None:
             self.attributes["inferentia"] = self.attributes["trainium"]
 
-        if not isinstance(BATCH_DEFAULT_TAGS, dict) and not all(
-            isinstance(k, str) and isinstance(v, str)
-            for k, v in BATCH_DEFAULT_TAGS.items()
-        ):
-            raise BatchException(
-                "BATCH_DEFAULT_TAGS environment variable must be Dict[str, str]"
-            )
+        if not isinstance(metaflow_config.BATCH_DEFAULT_TAGS, dict):
+            raise BatchException("BATCH_DEFAULT_TAGS environment variable must be Dict[str, str]")
+        if not all(isinstance(k, str) and isinstance(v, str) for k, v in metaflow_config.BATCH_DEFAULT_TAGS.items()):
+            raise BatchException("BATCH_DEFAULT_TAGS environment variable must be Dict[str, str]")
+
         if self.attributes["aws_batch_tags"] is None:
-            self.attributes["aws_batch_tags"] = BATCH_DEFAULT_TAGS
+            self.attributes["aws_batch_tags"] = metaflow_config.BATCH_DEFAULT_TAGS
 
         if self.attributes["aws_batch_tags"] is not None:
             if not isinstance(self.attributes["aws_batch_tags"], dict) and not all(
@@ -206,7 +204,7 @@ def __init__(self, attributes=None, statically_defined=False):
                 raise BatchException("aws_batch_tags must be Dict[str, str]")
 
             self.attributes["aws_batch_tags"] = {
-                **BATCH_DEFAULT_TAGS,
+                **metaflow_config.BATCH_DEFAULT_TAGS,
                 **self.attributes["aws_batch_tags"],
             }