Complete generator fix

Author: vaibhavatlan

pyatlan/generator/class_generator.py

@@ -227,6 +227,7 @@ class AssetInfo:
         "DataProduct",
         "Referenceable",
         "Purpose",
+        "alpha_DQRule",
     }
     _CORE_ASSETS = {
         "Referenceable",

pyatlan/generator/templates/methods/imports/alpha__d_q_rule.jinja2

@@ -0,0 +1,15 @@
+import time
+
+from pyatlan.model.enums import (
+    alpha_DQDimension,
+    alpha_DQResult,
+    alpha_DQRuleAlertPriority,
+    alpha_DQRuleStatus,
+    alpha_DQRuleThresholdCompareOperator,
+    alpha_DQRuleThresholdUnit,
+    alpha_DQSourceSyncStatus,
+)
+from pyatlan.model.structs import (
+    alpha_DQRuleConfigArguments,
+    alpha_DQRuleThresholdObject,
+)

pyatlan/generator/templates/module.jinja2

@@ -40,6 +40,9 @@ from .process import Process
 {% if asset_info.name in asset_info._ASSETS_REQUIRE_CLIENT %}
 if TYPE_CHECKING:
     from pyatlan.client.atlan import AtlanClient
+    {% if asset_info.name == 'alpha_DQRule' %}
+    from pyatlan.model.assets import Column
+    {% endif %}
 {% endif %}
 
 {% if asset_info.name == 'Asset' %}

pyatlan/generator/templates/structs.jinja2

@@ -14,7 +14,8 @@ from pyatlan.model.enums import (
     BadgeConditionColor,
     SourceCostUnitType,
     FormFieldDimension,
-    FormFieldType
+    FormFieldType,
+    alpha_DQRuleThresholdUnit,
 )
 from pyatlan.model.utils import to_camel_case
 from pyatlan.utils import select_optional_set_fields, validate_required_fields

pyatlan/model/assets/core/alpha__d_q_rule.py

@@ -12,6 +12,7 @@
 
 from pydantic.v1 import Field, validator
 
+from pyatlan.errors import ErrorCode
 from pyatlan.model.enums import (
     alpha_DQDimension,
     alpha_DQResult,
@@ -32,8 +33,9 @@
     alpha_DQRuleConfigArguments,
     alpha_DQRuleThresholdObject,
 )
-from pyatlan.utils import ErrorCode, init_guid, validate_required_fields
+from pyatlan.utils import init_guid, validate_required_fields
 
+from .asset import SelfAsset
 from .data_quality import DataQuality
 
 if TYPE_CHECKING:
@@ -1170,5 +1172,5 @@ def creator(
 
 
 from .alpha__d_q_rule_template import alpha_DQRuleTemplate  # noqa: E402, F401
-from .asset import Asset, SelfAsset  # noqa: E402, F401
+from .asset import Asset  # noqa: E402, F401
 # from .column import co