acryl-datahub-cloud 0.3.12rc8__py3-none-any.whl → 0.3.12rc10__py3-none-any.whl

acryl_datahub_cloud/_codegen_config.json

@@ -1,6 +1,6 @@
 {
   "name": "acryl-datahub-cloud",
-  "version": "0.3.12rc8",
+  "version": "0.3.12rc10",
   "install_requires": [
     "avro-gen3==0.7.16",
     "acryl-datahub"

acryl_datahub_cloud/datahub_reporting/datahub_form_reporting.py

@@ -10,7 +10,7 @@ from pydantic import BaseModel
 from acryl_datahub_cloud.elasticsearch.graph_service import BaseModelRow
 from datahub.emitter.mcp import MetadataChangeProposalWrapper
 from datahub.ingestion.graph.client import DataHubGraph
-from datahub.ingestion.graph.filters import RawSearchFilterRule
+from datahub.ingestion.graph.filters import RawSearchFilter
 from datahub.metadata.schema_classes import (
     DomainPropertiesClass,
     FormAssociationClass,
@@ -149,7 +149,7 @@ class DataHubFormReportingData(FormData):
             )
         )
 
-    def get_form_existence_or_filters(self) -> List[RawSearchFilterRule]:
+    def get_form_existence_or_filters(self) -> RawSearchFilter:
         """
         Datasets must either have completedForms or incompleteForms assigned to
         them
@@ -157,25 +157,41 @@ class DataHubFormReportingData(FormData):
         if self.allowed_forms:
             return [
                 {
-                    "field": "completedForms",
-                    "condition": "EQUAL",
-                    "values": self.allowed_forms,
+                    "and": [
+                        {
+                            "field": "completedForms",
+                            "condition": "EQUAL",
+                            "values": self.allowed_forms,
+                        }
+                    ]
                 },
                 {
-                    "field": "incompleteForms",
-                    "condition": "EQUAL",
-                    "values": self.allowed_forms,
+                    "and": [
+                        {
+                            "field": "incompleteForms",
+                            "condition": "EQUAL",
+                            "values": self.allowed_forms,
+                        }
+                    ]
                 },
             ]
         else:
             return [
                 {
-                    "field": "completedForms",
-                    "condition": "EXISTS",
+                    "and": [
+                        {
+                            "field": "completedForms",
+                            "condition": "EXISTS",
+                        }
+                    ]
                 },
                 {
-                    "field": "incompleteForms",
-                    "condition": "EXISTS",
+                    "and": [
+                        {
+                            "field": "incompleteForms",
+                            "condition": "EXISTS",
+                        }
+                    ]
                 },
             ]
 
@@ -293,7 +309,7 @@ class DataHubFormReportingData(FormData):
         extra_fields = [f for f in self.DataHubDatasetSearchRow.__fields__]
         # TODO: Replace with the new search/filter SDK.
         result = self.graph.get_results_by_filter(
-            extra_or_filters=[{"and": self.get_form_existence_or_filters()}],
+            extra_or_filters=self.get_form_existence_or_filters(),
             extra_source_fields=extra_fields,
             skip_cache=True,
         )

acryl_datahub_cloud/metadata/schema.avsc

@@ -14983,6 +14983,15 @@
         "doc": "Overall rank of the dataset based on the the last 30 days insert/update operation count on a platform\nIt returns one plus the number of rows proceeding or equals to the current row in the ordering. "
       },
       {
+        "Searchable": {
+          "/*": {
+            "addToFilters": true,
+            "fieldName": "topUsersLast30Days",
+            "fieldType": "URN",
+            "filterNameOverride": "Top users last 30 days",
+            "hasValuesFieldName": "hasTopUsersLast30Days"
+          }
+        },
         "Urn": "Urn",
         "urn_is_array": true,
         "type": [

acryl_datahub_cloud/metadata/schemas/UsageFeatures.avsc

@@ -160,6 +160,15 @@
       "doc": "Overall rank of the dataset based on the the last 30 days insert/update operation count on a platform\nIt returns one plus the number of rows proceeding or equals to the current row in the ordering. "
     },
     {
+      "Searchable": {
+        "/*": {
+          "addToFilters": true,
+          "fieldName": "topUsersLast30Days",
+          "fieldType": "URN",
+          "filterNameOverride": "Top users last 30 days",
+          "hasValuesFieldName": "hasTopUsersLast30Days"
+        }
+      },
       "type": [
         "null",
         {

acryl_datahub_cloud/sdk/assertion_input/assertion_input.py

@@ -911,9 +911,7 @@ class _AssertionInput(ABC):
         enabled: bool = True,
         schedule: Optional[Union[str, models.CronScheduleClass]] = None,
         detection_mechanism: DetectionMechanismInputTypes = None,
-        incident_behavior: Optional[
-            Union[AssertionIncidentBehavior, list[AssertionIncidentBehavior]]
-        ] = None,
+        incident_behavior: Optional[AssertionIncidentBehaviorInputTypes] = None,
         tags: Optional[TagsInputType] = None,
         source_type: str = models.AssertionSourceTypeClass.NATIVE,  # Verified on init to be a valid enum value
         created_by: Union[str, CorpUserUrn],
@@ -932,7 +930,11 @@ class _AssertionInput(ABC):
             display_name: The display name of the assertion. If not provided, a random display name will be generated.
             enabled: Whether the assertion is enabled. Defaults to True.
             detection_mechanism: The detection mechanism to be used for the assertion.
-            incident_behavior: The incident behavior to be applied to the assertion.
+            incident_behavior: The incident behavior to be applied to the assertion. Accepts:
+                - String values: "raise_on_fail", "resolve_on_pass"
+                - Enum values: AssertionIncidentBehavior.RAISE_ON_FAIL, AssertionIncidentBehavior.RESOLVE_ON_PASS
+                - Lists of any of the above values
+                - None (default behavior)
             tags: The tags to be applied to the assertion.
             source_type: The source type of the assertion. Defaults to models.AssertionSourceTypeClass.NATIVE.
             created_by: The actor that created the assertion.
@@ -1321,9 +1323,7 @@ class _SmartFreshnessAssertionInput(
         sensitivity: Optional[Union[str, InferenceSensitivity]] = None,
         exclusion_windows: Optional[ExclusionWindowInputTypes] = None,
         training_data_lookback_days: Optional[int] = None,
-        incident_behavior: Optional[
-            Union[AssertionIncidentBehavior, list[AssertionIncidentBehavior]]
-        ] = None,
+        incident_behavior: Optional[AssertionIncidentBehaviorInputTypes] = None,
         tags: Optional[TagsInputType] = None,
         created_by: Union[str, CorpUserUrn],
         created_at: datetime,
@@ -1504,9 +1504,7 @@ class _SmartVolumeAssertionInput(_AssertionInput, _HasSmartAssertionInputs):
         sensitivity: Optional[Union[str, InferenceSensitivity]] = None,
         exclusion_windows: Optional[ExclusionWindowInputTypes] = None,
         training_data_lookback_days: Optional[int] = None,
-        incident_behavior: Optional[
-            Union[AssertionIncidentBehavior, list[AssertionIncidentBehavior]]
-        ] = None,
+        incident_behavior: Optional[AssertionIncidentBehaviorInputTypes] = None,
         tags: Optional[TagsInputType] = None,
         created_by: Union[str, CorpUserUrn],
         created_at: datetime,

acryl_datahub_cloud/sdk/assertion_input/freshness_assertion_input.py

@@ -5,7 +5,7 @@ from acryl_datahub_cloud.sdk.assertion_input.assertion_input import (
     DEFAULT_DAILY_SCHEDULE,
     HIGH_WATERMARK_ALLOWED_FIELD_TYPES,
     LAST_MODIFIED_ALLOWED_FIELD_TYPES,
-    AssertionIncidentBehavior,
+    AssertionIncidentBehaviorInputTypes,
     DetectionMechanismInputTypes,
     FieldSpecType,
     TimeWindowSizeInputTypes,
@@ -49,9 +49,7 @@ class _FreshnessAssertionInput(_AssertionInput, _HasFreshnessFeatures):
         enabled: bool = True,
         schedule: Optional[Union[str, models.CronScheduleClass]] = None,
         detection_mechanism: DetectionMechanismInputTypes = None,
-        incident_behavior: Optional[
-            Union[AssertionIncidentBehavior, list[AssertionIncidentBehavior]]
-        ] = None,
+        incident_behavior: Optional[AssertionIncidentBehaviorInputTypes] = None,
         tags: Optional[TagsInputType] = None,
         created_by: Union[str, CorpUserUrn],
         created_at: datetime,

acryl_datahub_cloud/sdk/assertion_input/smart_column_metric_assertion_input.py

@@ -8,7 +8,7 @@ from acryl_datahub_cloud.sdk.assertion_input.assertion_input import (
     NO_PARAMETER_OPERATORS,
     RANGE_OPERATORS,
     SINGLE_VALUE_OPERATORS,
-    AssertionIncidentBehavior,
+    AssertionIncidentBehaviorInputTypes,
     AssertionInfoInputType,
     DetectionMechanismInputTypes,
     ExclusionWindowInputTypes,
@@ -282,9 +282,7 @@ class _SmartColumnMetricAssertionInput(_AssertionInput, _HasSmartAssertionInputs
         sensitivity: Optional[Union[str, InferenceSensitivity]] = None,
         exclusion_windows: Optional[ExclusionWindowInputTypes] = None,
         training_data_lookback_days: Optional[int] = None,
-        incident_behavior: Optional[
-            Union[AssertionIncidentBehavior, list[AssertionIncidentBehavior]]
-        ] = None,
+        incident_behavior: Optional[AssertionIncidentBehaviorInputTypes] = None,
         tags: Optional[TagsInputType] = None,
         created_by: Union[str, CorpUserUrn],
         created_at: datetime,
@@ -312,7 +310,7 @@ class _SmartColumnMetricAssertionInput(_AssertionInput, _HasSmartAssertionInputs
             sensitivity: The sensitivity of the assertion.
             exclusion_windows: The exclusion windows of the assertion.
             training_data_lookback_days: The training data lookback days of the assertion.
-            incident_behavior: The incident behavior of the assertion.
+            incident_behavior: The incident behavior of the assertion. Accepts strings, enum values, lists, or None.
             tags: The tags of the assertion.
             created_by: The creator of the assertion.
             created_at: The creation time of the assertion.

acryl_datahub_cloud/sdk/assertion_input/sql_assertion_input.py

@@ -8,7 +8,7 @@ from acryl_datahub_cloud.sdk.assertion_input.assertion_input import (
     DEFAULT_EVERY_SIX_HOURS_SCHEDULE,
     RANGE_OPERATORS,
     SINGLE_VALUE_NUMERIC_OPERATORS,
-    AssertionIncidentBehavior,
+    AssertionIncidentBehaviorInputTypes,
     FieldSpecType,
     _AssertionInput,
     _try_parse_and_validate_schema_classes_enum,
@@ -162,9 +162,7 @@ class _SqlAssertionInput(_AssertionInput):
         display_name: Optional[str] = None,
         enabled: bool = True,
         schedule: Optional[Union[str, models.CronScheduleClass]] = None,
-        incident_behavior: Optional[
-            Union[AssertionIncidentBehavior, list[AssertionIncidentBehavior]]
-        ] = None,
+        incident_behavior: Optional[AssertionIncidentBehaviorInputTypes] = None,
         tags: Optional[TagsInputType] = None,
         created_by: Union[str, CorpUserUrn],
         created_at: datetime,

acryl_datahub_cloud/sdk/assertion_input/volume_assertion_input.py

@@ -7,7 +7,7 @@ from pydantic import BaseModel, Extra
 
 from acryl_datahub_cloud.sdk.assertion_input.assertion_input import (
     DEFAULT_EVERY_SIX_HOURS_SCHEDULE,
-    AssertionIncidentBehavior,
+    AssertionIncidentBehaviorInputTypes,
     DetectionMechanismInputTypes,
     FieldSpecType,
     _AssertionInput,
@@ -530,9 +530,7 @@ class _VolumeAssertionInput(_AssertionInput):
         enabled: bool = True,
         schedule: Optional[Union[str, models.CronScheduleClass]] = None,
         detection_mechanism: DetectionMechanismInputTypes = None,
-        incident_behavior: Optional[
-            Union[AssertionIncidentBehavior, list[AssertionIncidentBehavior]]
-        ] = None,
+        incident_behavior: Optional[AssertionIncidentBehaviorInputTypes] = None,
         tags: Optional[TagsInputType] = None,
         created_by: Union[str, CorpUserUrn],
         created_at: datetime,