acryl-datahub-cloud 0.3.9.2rc1__tar.gz → 0.3.10rc0__tar.gz

{acryl_datahub_cloud-0.3.9.2rc1 → acryl_datahub_cloud-0.3.10rc0}/PKG-INFO

@@ -1,90 +1,90 @@
 Metadata-Version: 2.4
 Name: acryl-datahub-cloud
-Version: 0.3.9.2rc1
+Version: 0.3.10rc0
 Requires-Dist: avro-gen3==0.7.16
 Requires-Dist: acryl-datahub
 Provides-Extra: datahub-lineage-features
-Requires-Dist: pandas; extra == "datahub-lineage-features"
-Requires-Dist: opensearch-py==2.4.2; extra == "datahub-lineage-features"
 Requires-Dist: pyarrow; extra == "datahub-lineage-features"
 Requires-Dist: pydantic<2; extra == "datahub-lineage-features"
+Requires-Dist: opensearch-py==2.4.2; extra == "datahub-lineage-features"
+Requires-Dist: pandas; extra == "datahub-lineage-features"
 Requires-Dist: duckdb; extra == "datahub-lineage-features"
 Provides-Extra: datahub-reporting-forms
-Requires-Dist: pandas; extra == "datahub-reporting-forms"
-Requires-Dist: termcolor==2.5.0; extra == "datahub-reporting-forms"
 Requires-Dist: pyarrow; extra == "datahub-reporting-forms"
 Requires-Dist: pydantic<2; extra == "datahub-reporting-forms"
+Requires-Dist: termcolor==2.5.0; extra == "datahub-reporting-forms"
 Requires-Dist: boto3; extra == "datahub-reporting-forms"
+Requires-Dist: pandas; extra == "datahub-reporting-forms"
 Requires-Dist: duckdb; extra == "datahub-reporting-forms"
 Provides-Extra: datahub-reporting-extract-graph
-Requires-Dist: pandas; extra == "datahub-reporting-extract-graph"
-Requires-Dist: opensearch-py==2.4.2; extra == "datahub-reporting-extract-graph"
 Requires-Dist: pyarrow; extra == "datahub-reporting-extract-graph"
 Requires-Dist: pydantic<2; extra == "datahub-reporting-extract-graph"
+Requires-Dist: opensearch-py==2.4.2; extra == "datahub-reporting-extract-graph"
 Requires-Dist: boto3; extra == "datahub-reporting-extract-graph"
+Requires-Dist: pandas; extra == "datahub-reporting-extract-graph"
 Requires-Dist: duckdb; extra == "datahub-reporting-extract-graph"
 Provides-Extra: datahub-reporting-extract-sql
-Requires-Dist: pandas; extra == "datahub-reporting-extract-sql"
 Requires-Dist: pyarrow; extra == "datahub-reporting-extract-sql"
 Requires-Dist: pydantic<2; extra == "datahub-reporting-extract-sql"
 Requires-Dist: boto3; extra == "datahub-reporting-extract-sql"
+Requires-Dist: pandas; extra == "datahub-reporting-extract-sql"
 Requires-Dist: duckdb; extra == "datahub-reporting-extract-sql"
 Provides-Extra: datahub-usage-reporting
-Requires-Dist: pandas; extra == "datahub-usage-reporting"
-Requires-Dist: elasticsearch==7.13.4; extra == "datahub-usage-reporting"
+Requires-Dist: polars==1.23.0; extra == "datahub-usage-reporting"
+Requires-Dist: pydantic<2; extra == "datahub-usage-reporting"
 Requires-Dist: opensearch-py==2.4.2; extra == "datahub-usage-reporting"
-Requires-Dist: pyarrow; extra == "datahub-usage-reporting"
+Requires-Dist: termcolor==2.5.0; extra == "datahub-usage-reporting"
 Requires-Dist: boto3; extra == "datahub-usage-reporting"
 Requires-Dist: scipy<=1.14.1; extra == "datahub-usage-reporting"
-Requires-Dist: pyarrow<=18.0.0; extra == "datahub-usage-reporting"
-Requires-Dist: polars==1.23.0; extra == "datahub-usage-reporting"
-Requires-Dist: termcolor==2.5.0; extra == "datahub-usage-reporting"
-Requires-Dist: pydantic<2; extra == "datahub-usage-reporting"
+Requires-Dist: pandas; extra == "datahub-usage-reporting"
+Requires-Dist: pyarrow; extra == "datahub-usage-reporting"
+Requires-Dist: elasticsearch==7.13.4; extra == "datahub-usage-reporting"
 Requires-Dist: numpy<2; extra == "datahub-usage-reporting"
+Requires-Dist: pyarrow<=18.0.0; extra == "datahub-usage-reporting"
 Requires-Dist: duckdb; extra == "datahub-usage-reporting"
 Provides-Extra: datahub-metadata-sharing
 Requires-Dist: tenacity; extra == "datahub-metadata-sharing"
 Provides-Extra: acryl-cs-issues
+Requires-Dist: jinja2; extra == "acryl-cs-issues"
+Requires-Dist: zenpy; extra == "acryl-cs-issues"
 Requires-Dist: slack-sdk; extra == "acryl-cs-issues"
 Requires-Dist: openai; extra == "acryl-cs-issues"
-Requires-Dist: zenpy; extra == "acryl-cs-issues"
-Requires-Dist: jinja2; extra == "acryl-cs-issues"
 Provides-Extra: all
-Requires-Dist: zenpy; extra == "all"
-Requires-Dist: pyarrow; extra == "all"
-Requires-Dist: slack-sdk; extra == "all"
-Requires-Dist: boto3; extra == "all"
-Requires-Dist: pyarrow<=18.0.0; extra == "all"
-Requires-Dist: polars==1.23.0; extra == "all"
-Requires-Dist: termcolor==2.5.0; extra == "all"
 Requires-Dist: pydantic<2; extra == "all"
-Requires-Dist: duckdb; extra == "all"
-Requires-Dist: pandas; extra == "all"
-Requires-Dist: elasticsearch==7.13.4; extra == "all"
 Requires-Dist: opensearch-py==2.4.2; extra == "all"
-Requires-Dist: openai; extra == "all"
 Requires-Dist: scipy<=1.14.1; extra == "all"
+Requires-Dist: pandas; extra == "all"
+Requires-Dist: tenacity; extra == "all"
+Requires-Dist: duckdb; extra == "all"
+Requires-Dist: polars==1.23.0; extra == "all"
+Requires-Dist: slack-sdk; extra == "all"
+Requires-Dist: openai; extra == "all"
+Requires-Dist: termcolor==2.5.0; extra == "all"
+Requires-Dist: boto3; extra == "all"
 Requires-Dist: jinja2; extra == "all"
+Requires-Dist: zenpy; extra == "all"
+Requires-Dist: pyarrow; extra == "all"
+Requires-Dist: elasticsearch==7.13.4; extra == "all"
 Requires-Dist: numpy<2; extra == "all"
-Requires-Dist: tenacity; extra == "all"
+Requires-Dist: pyarrow<=18.0.0; extra == "all"
 Provides-Extra: dev
-Requires-Dist: pandas; extra == "dev"
-Requires-Dist: elasticsearch==7.13.4; extra == "dev"
+Requires-Dist: polars==1.23.0; extra == "dev"
+Requires-Dist: pydantic<2; extra == "dev"
 Requires-Dist: opensearch-py==2.4.2; extra == "dev"
-Requires-Dist: zenpy; extra == "dev"
-Requires-Dist: pyarrow; extra == "dev"
 Requires-Dist: slack-sdk; extra == "dev"
 Requires-Dist: openai; extra == "dev"
+Requires-Dist: termcolor==2.5.0; extra == "dev"
 Requires-Dist: boto3; extra == "dev"
+Requires-Dist: jinja2; extra == "dev"
 Requires-Dist: scipy<=1.14.1; extra == "dev"
-Requires-Dist: pyarrow<=18.0.0; extra == "dev"
-Requires-Dist: tenacity; extra == "dev"
-Requires-Dist: polars==1.23.0; extra == "dev"
+Requires-Dist: pandas; extra == "dev"
+Requires-Dist: zenpy; extra == "dev"
 Requires-Dist: acryl-datahub[dev]; extra == "dev"
-Requires-Dist: jinja2; extra == "dev"
-Requires-Dist: termcolor==2.5.0; extra == "dev"
-Requires-Dist: pydantic<2; extra == "dev"
+Requires-Dist: pyarrow; extra == "dev"
+Requires-Dist: elasticsearch==7.13.4; extra == "dev"
 Requires-Dist: numpy<2; extra == "dev"
+Requires-Dist: tenacity; extra == "dev"
+Requires-Dist: pyarrow<=18.0.0; extra == "dev"
 Requires-Dist: duckdb; extra == "dev"
 Dynamic: provides-extra
 Dynamic: requires-dist

{acryl_datahub_cloud-0.3.9.2rc1 → acryl_datahub_cloud-0.3.10rc0}/src/acryl_datahub_cloud/_codegen_config.json

@@ -1,6 +1,6 @@
 {
   "name": "acryl-datahub-cloud",
-  "version": "0.3.9.2rc1",
+  "version": "0.3.10rc0",
   "install_requires": [
     "avro-gen3==0.7.16",
     "acryl-datahub"

{acryl_datahub_cloud-0.3.9.2rc1 → acryl_datahub_cloud-0.3.10rc0}/src/acryl_datahub_cloud/metadata/_urns/urn_defs.py

@@ -1458,6 +1458,18 @@ class ChartUrn(_SpecificUrn):
     def from_key_aspect(cls, key_aspect: "ChartKeyClass") -> "ChartUrn":
         return cls(dashboard_tool=key_aspect.dashboardTool, chart_id=key_aspect.chartId)
 
+    @classmethod
+    def create_from_ids(
+        cls,
+        platform: str,
+        name: str,
+        platform_instance: Optional[str] = None,
+    ) -> "ChartUrn":
+        return ChartUrn(
+            dashboard_tool=platform,
+            chart_id=f"{platform_instance}.{name}" if platform_instance else name,
+        )
+        
     @property
     def dashboard_tool(self) -> str:
         return self._entity_ids[0]
@@ -1585,62 +1597,6 @@ class SubscriptionUrn(_SpecificUrn):
     def id(self) -> str:
         return self._entity_ids[0]
 
-if TYPE_CHECKING:
-    from datahub.metadata.schema_classes import AnomalyKeyClass
-
-class AnomalyUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[Literal["anomaly"]] = "anomaly"
-    _URN_PARTS: ClassVar[int] = 1
-
-    def __init__(self, id: Union["AnomalyUrn", str], *, _allow_coercion: bool = True) -> None:
-        if _allow_coercion:
-            # Field coercion logic (if any is required).
-            if isinstance(id, str):
-                if id.startswith('urn:li:'):
-                    try:
-                        id = AnomalyUrn.from_string(id)
-                    except InvalidUrnError:
-                        raise InvalidUrnError(f'Expecting a AnomalyUrn but got {id}')
-                else:
-                    id = UrnEncoder.encode_string(id)
-
-        # Validation logic.
-        if not id:
-            raise InvalidUrnError("AnomalyUrn id cannot be empty")
-        if isinstance(id, AnomalyUrn):
-            id = id.id
-        elif isinstance(id, Urn):
-            raise InvalidUrnError(f'Expecting a AnomalyUrn but got {id}')
-        if UrnEncoder.contains_reserved_char(id):
-            raise InvalidUrnError(f'AnomalyUrn id contains reserved characters')
-
-        super().__init__(self.ENTITY_TYPE, [id])
-
-    @classmethod
-    def _parse_ids(cls, entity_ids: List[str]) -> "AnomalyUrn":
-        if len(entity_ids) != cls._URN_PARTS:
-            raise InvalidUrnError(f"AnomalyUrn should have {cls._URN_PARTS} parts, got {len(entity_ids)}: {entity_ids}")
-        return cls(id=entity_ids[0], _allow_coercion=False)
-
-    @classmethod
-    def underlying_key_aspect_type(cls) -> Type["AnomalyKeyClass"]:
-        from datahub.metadata.schema_classes import AnomalyKeyClass
-
-        return AnomalyKeyClass
-
-    def to_key_aspect(self) -> "AnomalyKeyClass":
-        from datahub.metadata.schema_classes import AnomalyKeyClass
-
-        return AnomalyKeyClass(id=self.id)
-
-    @classmethod
-    def from_key_aspect(cls, key_aspect: "AnomalyKeyClass") -> "AnomalyUrn":
-        return cls(id=key_aspect.id)
-
-    @property
-    def id(self) -> str:
-        return self._entity_ids[0]
-
 if TYPE_CHECKING:
     from datahub.metadata.schema_classes import RemoteExecutorPoolKeyClass
 
@@ -2400,6 +2356,62 @@ class ConstraintUrn(_SpecificUrn):
     def id(self) -> str:
         return self._entity_ids[0]
 
+if TYPE_CHECKING:
+    from datahub.metadata.schema_classes import MonitorSuiteKeyClass
+
+class MonitorSuiteUrn(_SpecificUrn):
+    ENTITY_TYPE: ClassVar[Literal["monitorSuite"]] = "monitorSuite"
+    _URN_PARTS: ClassVar[int] = 1
+
+    def __init__(self, id: Union["MonitorSuiteUrn", str], *, _allow_coercion: bool = True) -> None:
+        if _allow_coercion:
+            # Field coercion logic (if any is required).
+            if isinstance(id, str):
+                if id.startswith('urn:li:'):
+                    try:
+                        id = MonitorSuiteUrn.from_string(id)
+                    except InvalidUrnError:
+                        raise InvalidUrnError(f'Expecting a MonitorSuiteUrn but got {id}')
+                else:
+                    id = UrnEncoder.encode_string(id)
+
+        # Validation logic.
+        if not id:
+            raise InvalidUrnError("MonitorSuiteUrn id cannot be empty")
+        if isinstance(id, MonitorSuiteUrn):
+            id = id.id
+        elif isinstance(id, Urn):
+            raise InvalidUrnError(f'Expecting a MonitorSuiteUrn but got {id}')
+        if UrnEncoder.contains_reserved_char(id):
+            raise InvalidUrnError(f'MonitorSuiteUrn id contains reserved characters')
+
+        super().__init__(self.ENTITY_TYPE, [id])
+
+    @classmethod
+    def _parse_ids(cls, entity_ids: List[str]) -> "MonitorSuiteUrn":
+        if len(entity_ids) != cls._URN_PARTS:
+            raise InvalidUrnError(f"MonitorSuiteUrn should have {cls._URN_PARTS} parts, got {len(entity_ids)}: {entity_ids}")
+        return cls(id=entity_ids[0], _allow_coercion=False)
+
+    @classmethod
+    def underlying_key_aspect_type(cls) -> Type["MonitorSuiteKeyClass"]:
+        from datahub.metadata.schema_classes import MonitorSuiteKeyClass
+
+        return MonitorSuiteKeyClass
+
+    def to_key_aspect(self) -> "MonitorSuiteKeyClass":
+        from datahub.metadata.schema_classes import MonitorSuiteKeyClass
+
+        return MonitorSuiteKeyClass(id=self.id)
+
+    @classmethod
+    def from_key_aspect(cls, key_aspect: "MonitorSuiteKeyClass") -> "MonitorSuiteUrn":
+        return cls(id=key_aspect.id)
+
+    @property
+    def id(self) -> str:
+        return self._entity_ids[0]
+
 if TYPE_CHECKING:
     from datahub.metadata.schema_classes import GlossaryNodeKeyClass
 
@@ -2793,6 +2805,18 @@ class DashboardUrn(_SpecificUrn):
     def from_key_aspect(cls, key_aspect: "DashboardKeyClass") -> "DashboardUrn":
         return cls(dashboard_tool=key_aspect.dashboardTool, dashboard_id=key_aspect.dashboardId)
 
+    @classmethod
+    def create_from_ids(
+        cls,
+        platform: str,
+        name: str,
+        platform_instance: Optional[str] = None,
+    ) -> "DashboardUrn":
+        return DashboardUrn(
+            dashboard_tool=platform,
+            dashboard_id=f"{platform_instance}.{name}" if platform_instance else name,
+        )
+        
     @property
     def dashboard_tool(self) -> str:
         return self._entity_ids[0]

{acryl_datahub_cloud-0.3.9.2rc1 → acryl_datahub_cloud-0.3.10rc0}/src/acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/anomaly/__init__.py

@@ -7,27 +7,17 @@
 # pylint: skip-file
 # fmt: off
 # isort: skip_file
-from .....schema_classes import AnomalyInfoClass
-from .....schema_classes import AnomalyReviewClass
 from .....schema_classes import AnomalyReviewStateClass
 from .....schema_classes import AnomalySourceClass
 from .....schema_classes import AnomalySourcePropertiesClass
 from .....schema_classes import AnomalySourceTypeClass
-from .....schema_classes import AnomalyStateClass
-from .....schema_classes import AnomalyStatusClass
-from .....schema_classes import AnomalyStatusPropertiesClass
-from .....schema_classes import AnomalyTypeClass
+from .....schema_classes import MonitorAnomalyEventClass
 
 
-AnomalyInfo = AnomalyInfoClass
-AnomalyReview = AnomalyReviewClass
 AnomalyReviewState = AnomalyReviewStateClass
 AnomalySource = AnomalySourceClass
 AnomalySourceProperties = AnomalySourcePropertiesClass
 AnomalySourceType = AnomalySourceTypeClass
-AnomalyState = AnomalyStateClass
-AnomalyStatus = AnomalyStatusClass
-AnomalyStatusProperties = AnomalyStatusPropertiesClass
-AnomalyType = AnomalyTypeClass
+MonitorAnomalyEvent = MonitorAnomalyEventClass
 
 # fmt: on

{acryl_datahub_cloud-0.3.9.2rc1 → acryl_datahub_cloud-0.3.10rc0}/src/acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/assertion/__init__.py

@@ -15,8 +15,11 @@ from .....schema_classes import AssertionAdjustmentSettingsClass
 from .....schema_classes import AssertionAnalyticsRunEventClass
 from .....schema_classes import AssertionDryRunEventClass
 from .....schema_classes import AssertionDryRunResultClass
+from .....schema_classes import AssertionExclusionWindowClass
+from .....schema_classes import AssertionExclusionWindowTypeClass
 from .....schema_classes import AssertionInferenceDetailsClass
 from .....schema_classes import AssertionInfoClass
+from .....schema_classes import AssertionMonitorSensitivityClass
 from .....schema_classes import AssertionResultClass
 from .....schema_classes import AssertionResultErrorClass
 from .....schema_classes import AssertionResultErrorTypeClass
@@ -77,8 +80,11 @@ AssertionAdjustmentSettings = AssertionAdjustmentSettingsClass
 AssertionAnalyticsRunEvent = AssertionAnalyticsRunEventClass
 AssertionDryRunEvent = AssertionDryRunEventClass
 AssertionDryRunResult = AssertionDryRunResultClass
+AssertionExclusionWindow = AssertionExclusionWindowClass
+AssertionExclusionWindowType = AssertionExclusionWindowTypeClass
 AssertionInferenceDetails = AssertionInferenceDetailsClass
 AssertionInfo = AssertionInfoClass
+AssertionMonitorSensitivity = AssertionMonitorSensitivityClass
 AssertionResult = AssertionResultClass
 AssertionResultError = AssertionResultErrorClass
 AssertionResultErrorType = AssertionResultErrorTypeClass

{acryl_datahub_cloud-0.3.9.2rc1 → acryl_datahub_cloud-0.3.10rc0}/src/acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/common/__init__.py

@@ -54,6 +54,7 @@ from .....schema_classes import InputFieldClass
 from .....schema_classes import InputFieldsClass
 from .....schema_classes import InstitutionalMemoryClass
 from .....schema_classes import InstitutionalMemoryMetadataClass
+from .....schema_classes import InstitutionalMemoryMetadataSettingsClass
 from .....schema_classes import MLFeatureDataTypeClass
 from .....schema_classes import MediaClass
 from .....schema_classes import MediaTypeClass
@@ -141,6 +142,7 @@ InputField = InputFieldClass
 InputFields = InputFieldsClass
 InstitutionalMemory = InstitutionalMemoryClass
 InstitutionalMemoryMetadata = InstitutionalMemoryMetadataClass
+InstitutionalMemoryMetadataSettings = InstitutionalMemoryMetadataSettingsClass
 MLFeatureDataType = MLFeatureDataTypeClass
 Media = MediaClass
 MediaType = MediaTypeClass

{acryl_datahub_cloud-0.3.9.2rc1 → acryl_datahub_cloud-0.3.10rc0}/src/acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/metadata/key/__init__.py

@@ -55,6 +55,7 @@ from ......schema_classes import MLModelGroupKeyClass
 from ......schema_classes import MLModelKeyClass
 from ......schema_classes import MLPrimaryKeyKeyClass
 from ......schema_classes import MonitorKeyClass
+from ......schema_classes import MonitorSuiteKeyClass
 from ......schema_classes import NotebookKeyClass
 from ......schema_classes import OwnershipTypeKeyClass
 from ......schema_classes import PostKeyClass
@@ -120,6 +121,7 @@ MLModelGroupKey = MLModelGroupKeyClass
 MLModelKey = MLModelKeyClass
 MLPrimaryKeyKey = MLPrimaryKeyKeyClass
 MonitorKey = MonitorKeyClass
+MonitorSuiteKey = MonitorSuiteKeyClass
 NotebookKey = NotebookKeyClass
 OwnershipTypeKey = OwnershipTypeKeyClass
 PostKey = PostKeyClass

{acryl_datahub_cloud-0.3.9.2rc1 → acryl_datahub_cloud-0.3.10rc0}/src/acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/monitor/__init__.py

@@ -25,9 +25,17 @@ from .....schema_classes import DatasetSchemaSourceTypeClass
 from .....schema_classes import DatasetVolumeAssertionParametersClass
 from .....schema_classes import DatasetVolumeSourceTypeClass
 from .....schema_classes import EmbeddedAssertionClass
+from .....schema_classes import MonitorErrorClass
+from .....schema_classes import MonitorErrorTypeClass
 from .....schema_classes import MonitorInfoClass
 from .....schema_classes import MonitorModeClass
+from .....schema_classes import MonitorStateClass
 from .....schema_classes import MonitorStatusClass
+from .....schema_classes import MonitorSuiteAssertionSettingsClass
+from .....schema_classes import MonitorSuiteAssignmentSpecClass
+from .....schema_classes import MonitorSuiteInfoClass
+from .....schema_classes import MonitorSuiteNotificationSettingsClass
+from .....schema_classes import MonitorSuiteTypeClass
 from .....schema_classes import MonitorTimeseriesStateClass
 from .....schema_classes import MonitorTypeClass
 
@@ -50,9 +58,17 @@ DatasetSchemaSourceType = DatasetSchemaSourceTypeClass
 DatasetVolumeAssertionParameters = DatasetVolumeAssertionParametersClass
 DatasetVolumeSourceType = DatasetVolumeSourceTypeClass
 EmbeddedAssertion = EmbeddedAssertionClass
+MonitorError = MonitorErrorClass
+MonitorErrorType = MonitorErrorTypeClass
 MonitorInfo = MonitorInfoClass
 MonitorMode = MonitorModeClass
+MonitorState = MonitorStateClass
 MonitorStatus = MonitorStatusClass
+MonitorSuiteAssertionSettings = MonitorSuiteAssertionSettingsClass
+MonitorSuiteAssignmentSpec = MonitorSuiteAssignmentSpecClass
+MonitorSuiteInfo = MonitorSuiteInfoClass
+MonitorSuiteNotificationSettings = MonitorSuiteNotificationSettingsClass
+MonitorSuiteType = MonitorSuiteTypeClass
 MonitorTimeseriesState = MonitorTimeseriesStateClass
 MonitorType = MonitorTypeClass
 

{acryl_datahub_cloud-0.3.9.2rc1 → acryl_datahub_cloud-0.3.10rc0}/src/acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/timeseries/__init__.py

@@ -7,17 +7,25 @@
 # pylint: skip-file
 # fmt: off
 # isort: skip_file
+from .....schema_classes import AbsoluteTimeWindowClass
 from .....schema_classes import CalendarIntervalClass
+from .....schema_classes import DayOfWeekClass
+from .....schema_classes import HolidayWindowClass
 from .....schema_classes import PartitionSpecClass
 from .....schema_classes import PartitionTypeClass
 from .....schema_classes import TimeWindowClass
 from .....schema_classes import TimeWindowSizeClass
+from .....schema_classes import WeeklyWindowClass
 
 
+AbsoluteTimeWindow = AbsoluteTimeWindowClass
 CalendarInterval = CalendarIntervalClass
+DayOfWeek = DayOfWeekClass
+HolidayWindow = HolidayWindowClass
 PartitionSpec = PartitionSpecClass
 PartitionType = PartitionTypeClass
 TimeWindow = TimeWindowClass
 TimeWindowSize = TimeWindowSizeClass
+WeeklyWindow = WeeklyWindowClass
 
 # fmt: on