feat: Switch to Pkl-generated __init__.py and .pyi, remove _generate_pkg_init.py

- __init__.py and __init__.pyi are now generated directly by PythonMsgspecRenderer
  using lazy_loader.attach() with aggregated typedef data
- Removed _generate_pkg_init.py post-sync script (no longer needed)
- Updated skill to remove init generation step and __init__ rsync excludes
- Updated release notes for 9.2.0

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: Aryamanz29

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

@@ -45,7 +45,7 @@ cd "$MODELS_DIR"
 STAGING_DIR="$(mktemp -d)"
 OVERLAYS_PATH="${SDK_DIR}/pyatlan_v9/model/assets/_overlays/"
 
-pkl eval typedefs/*.pkl -m "$STAGING_DIR" -p sdkOnly=true -p sdk=true \
+pkl eval typedefs/*.pkl -m "$STAGING_DIR" -p sdkOnly=true \
   -p targetOutputDir=pyatlan_v9/model/assets/ \
   -p internalPackage=pyatlan_v9.model \
   -p sdkOverlaysBasePath="$OVERLAYS_PATH"
@@ -61,8 +61,6 @@ Copy generated files from the staging dir to the SDK, **excluding** these files
 
 | File | Reason |
 |------|--------|
-| `__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,8 +73,6 @@ 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' \
@@ -104,20 +100,7 @@ rm -rf "$STAGING_DIR"
     """Attributes of the relationship itself (e.g., description, status, etc.)."""
 ```
 
-### 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
+### 5. Run ruff auto-fix and format
 
 After syncing and patching, run ruff to fix unused imports and format the generated files:
 
@@ -127,13 +110,13 @@ uv run ruff check --fix --select F401,F811 pyatlan_v9/
 uv run ruff format pyatlan_v9/
 ```
 
-### 7. Run tests (if args contain "test")
+### 6. Run tests (if args contain "test")
 
 ```bash
 cd "${SDK_DIR}" && python -m pytest tests_v9/unit/ -x -q
 ```
 
-### 8. Report summary
+### 7. Report summary
 
 Report: how many files were generated, how many synced, how many excluded, and test results if applicable.
 

HISTORY.md

@@ -4,15 +4,10 @@
 
 - **Policy duplicate detection during retries**: When a policy creation request times out but actually succeeds on the backend, the SDK now detects the existing policy instead of blindly retrying and creating a duplicate. On retry, the transport layer searches for an existing `AuthPolicy` with the same name and persona GUID — if found, it returns a mock response containing the existing policy. Code that relied on retries always creating new policies will now receive the existing policy instead.
 
-### New Features
-
-- **AtlanTag semantic support for bulk operations**: `client.asset.save()` now supports `SaveSemantic` (APPEND, REMOVE, REPLACE) when saving entities with Atlan tags in bulk. The SDK determines the appropriate API flags (`appendTags`, `replaceTags`) and makes a single bulk API call. Mixed semantics in a single call are rejected by the backend.
-- **`KeywordTextDelimitedField`**: New field type for assets with delimiter-based text search support (using `atlan_text_analyzer_v2` analyzer).
-
 ### Experimental: `pyatlan_v9`
 
-- **IDE autocompletion**: Replaced custom AST-based lazy loading with `lazy_loader` + auto-generated `.pyi` stub file. All 864 asset classes now have full IDE type hints and autocompletion while keeping lazy import performance.
-- **Automated model sync**: Added `_generate_pkg_init.py` post-sync script and updated the GitHub Actions workflow to auto-generate `__init__.py` / `__init__.pyi` after Pkl model sync.
+- **IDE autocompletion**: Replaced custom AST-based lazy loading with `lazy_loader` + Pkl-generated `.pyi` stub file. All 860+ asset classes now have full IDE type hints and autocompletion while keeping lazy import performance.
+- **Pkl-generated `__init__.py` and `__init__.pyi`**: The Pkl renderer now aggregates all typedefs (via `import*` glob) to generate `__init__.py` (with `lazy_loader.attach()` mapping) and `__init__.pyi` (with explicit re-exports for IDEs) directly — no post-sync Python script needed. Hand-written types from `_init_manual.py` are included by reading the file at generation time.
 - **Model updates**: Regenerated all v9 models — GTC anchor regularization, entity lineage fields (`depth`, `immediate_upstream`, `immediate_downstream`), QuickSight `useLocalTypeAsPrefix`, fully Pkl-generated `DataQualityRule`, and referenceable field descriptors via overlays.
 - **`type_name` serialization fix**: `type_name` field default changed to `UNSET` so `omit_defaults=True` never omits `typeName` from API requests.
 - **Simplified `transform.py`**: Removed `_normalize_camel_key()` and related camelCase abbreviation handling — msgspec structs use explicit `field(name=...)` mappings. Asset registration now uses `cls.__name__` directly.

pyatlan/client/common/transport.py

@@ -17,7 +17,7 @@
 
 from pyatlan.client.constants import BULK_UPDATE, INDEX_SEARCH
 from pyatlan.errors import ErrorCode
-from pyatlan.model.search import Bool, DSL, IndexSearchRequest, Term
+from pyatlan.model.search import DSL, Bool, IndexSearchRequest, Term
 
 logger = logging.getLogger(__name__)
 

pyatlan_v9/model/assets/__init__.py

@@ -1,4 +1,4 @@
-# Auto-generated by _generate_pkg_init.py — DO NOT EDIT
+# Auto-generated by PythonMsgspecRenderer.pkl — DO NOT EDIT
 # SPDX-License-Identifier: Apache-2.0
 # Copyright 2024 Atlan Pte. Ltd.
 # isort: skip_file
@@ -520,23 +520,23 @@
     "_init_manual": [
         "AccessControl",
         "AuthPolicy",
-        "AzureEventHubConsumerGroup",
         "AzureEventHub",
+        "AzureEventHubConsumerGroup",
         "Badge",
         "BadgeCondition",
         "Cognite3DModel",
         "DataContract",
         "Persona",
         "Purpose",
+        "RelatedSuperset",
+        "RelatedSupersetChart",
+        "RelatedSupersetDashboard",
+        "RelatedSupersetDataset",
         "SnowflakeDynamicTable",
         "Superset",
         "SupersetChart",
         "SupersetDashboard",
         "SupersetDataset",
-        "RelatedSuperset",
-        "RelatedSupersetChart",
-        "RelatedSupersetDashboard",
-        "RelatedSupersetDataset",
     ],
     "_init_matillion": [
         "Matillion",
@@ -743,13 +743,7 @@
         "RelatedRedashQuery",
         "RelatedRedashVisualization",
     ],
-    "_init_referenceable": [
-        "AtlasClassification",
-        "Entity",
-        "Referenceable",
-        "RelatedReferenceable",
-        "TermAssignment",
-    ],
+    "_init_referenceable": ["Referenceable", "RelatedReferenceable"],
     "_init_resource": [
         "File",
         "Link",

pyatlan_v9/model/assets/__init__.pyi

@@ -1,4 +1,4 @@
-# Auto-generated by _generate_pkg_init.py — DO NOT EDIT
+# Auto-generated by PythonMsgspecRenderer.pkl — DO NOT EDIT
 # SPDX-License-Identifier: Apache-2.0
 # Copyright 2024 Atlan Pte. Ltd.
 
@@ -442,23 +442,23 @@ __all__ = [
     "RelatedLookerView",
     "AccessControl",
     "AuthPolicy",
-    "AzureEventHubConsumerGroup",
     "AzureEventHub",
+    "AzureEventHubConsumerGroup",
     "Badge",
     "BadgeCondition",
     "Cognite3DModel",
     "DataContract",
     "Persona",
     "Purpose",
+    "RelatedSuperset",
+    "RelatedSupersetChart",
+    "RelatedSupersetDashboard",
+    "RelatedSupersetDataset",
     "SnowflakeDynamicTable",
     "Superset",
     "SupersetChart",
     "SupersetDashboard",
     "SupersetDataset",
-    "RelatedSuperset",
-    "RelatedSupersetChart",
-    "RelatedSupersetDashboard",
-    "RelatedSupersetDataset",
     "Matillion",
     "MatillionComponent",
     "MatillionGroup",
@@ -635,11 +635,8 @@ __all__ = [
     "RelatedRedashDashboard",
     "RelatedRedashQuery",
     "RelatedRedashVisualization",
-    "AtlasClassification",
-    "Entity",
     "Referenceable",
     "RelatedReferenceable",
-    "TermAssignment",
     "File",
     "Link",
     "Readme",
@@ -1051,7 +1048,7 @@ from .cloud_related import RelatedAzure as RelatedAzure
 from .cloud_related import RelatedCloud as RelatedCloud
 from .cloud_related import RelatedGoogle as RelatedGoogle
 from .cognite import Cognite as Cognite
-from .cognite3_d_model import Cognite3DModel as Cognite3DModel
+from .cognite3d_model import Cognite3DModel as Cognite3DModel
 from .cognite_asset import CogniteAsset as CogniteAsset
 from .cognite_event import CogniteEvent as CogniteEvent
 from .cognite_file import CogniteFile as CogniteFile