Merge pull request #836 from atlanhq/BLDX-711

BLDX-711 |  Regenerate v9 models, fix tests, add lazy_loader + IDE stubs

Author: Aryamanz29

.claude/skills/generate-v9-models/SKILL.md

@@ -61,7 +61,8 @@ Copy generated files from the staging dir to the SDK, **excluding** these files
 
 | File | Reason |
 |------|--------|
-| `__init__.py` | Hand-written init with `__all__` |
+| `__init__.py` | Generated post-sync by `_generate_pkg_init.py` |
+| `__init__.pyi` | Generated post-sync by `_generate_pkg_init.py` |
 | `entity.py` | Patched: `_metadata_proxies`, `type_name: Any`, `SaveSemantic` |
 | `referenceable.py` | Patched: `InternalKeywordField`, field descriptors, helper exports |
 | `atlas_glossary.py` | Patched: GTC anchor-in-attributes handling |
@@ -75,6 +76,7 @@ Copy generated files from the staging dir to the SDK, **excluding** these files
 ```bash
 rsync -av \
   --exclude='__init__.py' \
+  --exclude='__init__.pyi' \
   --exclude='entity.py' \
   --exclude='referenceable.py' \
   --exclude='atlas_glossary.py' \
@@ -102,7 +104,20 @@ rm -rf "$STAGING_DIR"
     """Attributes of the relationship itself (e.g., description, status, etc.)."""
 ```
 
-### 5. Run ruff auto-fix and format
+### 5. Regenerate `__init__.py` and `__init__.pyi`
+
+After syncing, regenerate the package init files from the `_init_*.py` modules:
+
+```bash
+cd "${SDK_DIR}"
+uv run python pyatlan_v9/model/assets/_generate_pkg_init.py
+```
+
+This scans all `_init_*.py` files and generates:
+- `__init__.py` with `lazy_loader.attach()` for lazy imports
+- `__init__.pyi` with explicit imports for IDE type hints
+
+### 6. Run ruff auto-fix and format
 
 After syncing and patching, run ruff to fix unused imports and format the generated files:
 
@@ -112,13 +127,13 @@ uv run ruff check --fix --select F401,F811 pyatlan_v9/
 uv run ruff format pyatlan_v9/
 ```
 
-### 6. Run tests (if args contain "test")
+### 7. Run tests (if args contain "test")
 
 ```bash
 cd "${SDK_DIR}" && python -m pytest tests_v9/unit/ -x -q
 ```
 
-### 7. Report summary
+### 8. Report summary
 
 Report: how many files were generated, how many synced, how many excluded, and test results if applicable.
 

pyatlan/generator/class_generator.py

@@ -11,9 +11,9 @@
 import json
 import os
 import re
+from collections.abc import Iterable
 from enum import Enum
 from pathlib import Path
-from collections.abc import Iterable
 from typing import Any, Dict, List, NamedTuple, Optional, Set
 
 import networkx as nx

pyatlan_v9/client/aio/asset.py

@@ -2073,9 +2073,9 @@ async def add_dq_rule_schedule(
         updated_asset = asset_type.updater(
             qualified_name=asset_qualified_name, name=asset_name
         )
-        updated_asset.asset_d_q_schedule_time_zone = schedule_time_zone
-        updated_asset.asset_d_q_schedule_crontab = schedule_crontab
-        updated_asset.asset_d_q_schedule_type = DataQualityScheduleType.CRON
+        updated_asset.asset_dq_schedule_time_zone = schedule_time_zone
+        updated_asset.asset_dq_schedule_crontab = schedule_crontab
+        updated_asset.asset_dq_schedule_type = DataQualityScheduleType.CRON
         return await self.save(updated_asset)
 
     @validate_arguments
@@ -2099,7 +2099,7 @@ async def set_dq_row_scope_filter_column(
         updated_asset = asset_type.updater(
             qualified_name=asset_qualified_name, name=asset_name
         )
-        updated_asset.asset_d_q_row_scope_filter_column_qualified_name = (
+        updated_asset.asset_dq_row_scope_filter_column_qualified_name = (
             row_scope_filter_column_qualified_name
         )
         return await self.save(updated_asset)

pyatlan_v9/client/asset.py

@@ -2234,9 +2234,9 @@ def add_dq_rule_schedule(
         updated_asset = asset_type.updater(
             qualified_name=asset_qualified_name, name=asset_name
         )
-        updated_asset.asset_d_q_schedule_time_zone = schedule_time_zone
-        updated_asset.asset_d_q_schedule_crontab = schedule_crontab
-        updated_asset.asset_d_q_schedule_type = DataQualityScheduleType.CRON
+        updated_asset.asset_dq_schedule_time_zone = schedule_time_zone
+        updated_asset.asset_dq_schedule_crontab = schedule_crontab
+        updated_asset.asset_dq_schedule_type = DataQualityScheduleType.CRON
         return self.save(updated_asset)
 
     @validate_arguments
@@ -2260,7 +2260,7 @@ def set_dq_row_scope_filter_column(
         updated_asset = asset_type.updater(
             qualified_name=asset_qualified_name, name=asset_name
         )
-        updated_asset.asset_d_q_row_scope_filter_column_qualified_name = (
+        updated_asset.asset_dq_row_scope_filter_column_qualified_name = (
             row_scope_filter_column_qualified_name
         )
         return self.save(updated_asset)

pyatlan_v9/model/__init__.py

@@ -3,10 +3,19 @@
 
 from __future__ import annotations
 
-# Re-export all asset models from the assets subpackage
+from typing import Any
+
+# Re-export eagerly loaded base classes from the assets subpackage
 from .assets import *  # noqa: F401, F403
 
 # Re-export all names from the assets __all__
 from .assets import __all__ as _assets_all
 
 __all__ = list(_assets_all)
+
+
+def __getattr__(name: str) -> Any:
+    """Delegate lazy imports to the assets subpackage."""
+    from . import assets
+
+    return getattr(assets, name)