[gen/fix] Fixed pyatlan 6 changes in the templates

Author: Aryamanz29

pyatlan/generator/templates/methods/attribute/badge.jinja2

@@ -13,10 +13,11 @@
                 ["name", "cm_name", "cm_attribute", "badge_conditions"],
                 [name, cm_name, cm_attribute, badge_conditions],
             )
-            from pyatlan.cache.custom_metadata_cache import CustomMetadataCache
+            from pyatlan.client.atlan import AtlanClient
 
-            cm_id = CustomMetadataCache.get_id_for_name(cm_name)
-            cm_attr_id = CustomMetadataCache.get_attr_id_for_name(
+            client = AtlanClient.get_current_client()
+            cm_id = client.custom_metadata_cache.get_id_for_name(cm_name)
+            cm_attr_id = client.custom_metadata_cache.get_attr_id_for_name(
                 set_name=cm_name, attr_name=cm_attribute
             )
             return Badge.Attributes(

pyatlan/generator/templates/referenceable_methods.jinja2

@@ -59,12 +59,13 @@
 
     @property
     def atlan_tag_names(self) -> List[str]:
-        from pyatlan.cache.atlan_tag_cache import AtlanTagCache
+        from pyatlan.client.atlan import AtlanClient
         from pyatlan.model.constants import DELETED_
 
         if self.classification_names:
             return [
-                AtlanTagCache.get_name_for_id(tag_id) or DELETED_
+                AtlanClient.get_current_client().atlan_tag_cache.get_name_for_id(tag_id)
+                or DELETED_
                 for tag_id in self.classification_names
             ]
         return []