pulumi-databricks 1.77.0a1760375482__py3-none-any.whl → 1.79.0a1762839813__py3-none-any.whl

pulumi_databricks/feature_engineering_materialized_feature.py

@@ -0,0 +1,397 @@
+# coding=utf-8
+# *** WARNING: this file was generated by pulumi-language-python. ***
+# *** Do not edit by hand unless you're certain you know what you are doing! ***
+
+import builtins as _builtins
+import warnings
+import sys
+import pulumi
+import pulumi.runtime
+from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict, TypeAlias
+else:
+    from typing_extensions import NotRequired, TypedDict, TypeAlias
+from . import _utilities
+from . import outputs
+from ._inputs import *
+
+__all__ = ['FeatureEngineeringMaterializedFeatureArgs', 'FeatureEngineeringMaterializedFeature']
+
+@pulumi.input_type
+class FeatureEngineeringMaterializedFeatureArgs:
+    def __init__(__self__, *,
+                 feature_name: pulumi.Input[_builtins.str],
+                 offline_store_config: Optional[pulumi.Input['FeatureEngineeringMaterializedFeatureOfflineStoreConfigArgs']] = None,
+                 online_store_config: Optional[pulumi.Input['FeatureEngineeringMaterializedFeatureOnlineStoreConfigArgs']] = None,
+                 pipeline_schedule_state: Optional[pulumi.Input[_builtins.str]] = None):
+        """
+        The set of arguments for constructing a FeatureEngineeringMaterializedFeature resource.
+        :param pulumi.Input[_builtins.str] feature_name: The full name of the feature in Unity Catalog
+        :param pulumi.Input[_builtins.str] pipeline_schedule_state: The schedule state of the materialization pipeline. Possible values are: `ACTIVE`, `PAUSED`, `SNAPSHOT`
+        """
+        pulumi.set(__self__, "feature_name", feature_name)
+        if offline_store_config is not None:
+            pulumi.set(__self__, "offline_store_config", offline_store_config)
+        if online_store_config is not None:
+            pulumi.set(__self__, "online_store_config", online_store_config)
+        if pipeline_schedule_state is not None:
+            pulumi.set(__self__, "pipeline_schedule_state", pipeline_schedule_state)
+
+    @_builtins.property
+    @pulumi.getter(name="featureName")
+    def feature_name(self) -> pulumi.Input[_builtins.str]:
+        """
+        The full name of the feature in Unity Catalog
+        """
+        return pulumi.get(self, "feature_name")
+
+    @feature_name.setter
+    def feature_name(self, value: pulumi.Input[_builtins.str]):
+        pulumi.set(self, "feature_name", value)
+
+    @_builtins.property
+    @pulumi.getter(name="offlineStoreConfig")
+    def offline_store_config(self) -> Optional[pulumi.Input['FeatureEngineeringMaterializedFeatureOfflineStoreConfigArgs']]:
+        return pulumi.get(self, "offline_store_config")
+
+    @offline_store_config.setter
+    def offline_store_config(self, value: Optional[pulumi.Input['FeatureEngineeringMaterializedFeatureOfflineStoreConfigArgs']]):
+        pulumi.set(self, "offline_store_config", value)
+
+    @_builtins.property
+    @pulumi.getter(name="onlineStoreConfig")
+    def online_store_config(self) -> Optional[pulumi.Input['FeatureEngineeringMaterializedFeatureOnlineStoreConfigArgs']]:
+        return pulumi.get(self, "online_store_config")
+
+    @online_store_config.setter
+    def online_store_config(self, value: Optional[pulumi.Input['FeatureEngineeringMaterializedFeatureOnlineStoreConfigArgs']]):
+        pulumi.set(self, "online_store_config", value)
+
+    @_builtins.property
+    @pulumi.getter(name="pipelineScheduleState")
+    def pipeline_schedule_state(self) -> Optional[pulumi.Input[_builtins.str]]:
+        """
+        The schedule state of the materialization pipeline. Possible values are: `ACTIVE`, `PAUSED`, `SNAPSHOT`
+        """
+        return pulumi.get(self, "pipeline_schedule_state")
+
+    @pipeline_schedule_state.setter
+    def pipeline_schedule_state(self, value: Optional[pulumi.Input[_builtins.str]]):
+        pulumi.set(self, "pipeline_schedule_state", value)
+
+
+@pulumi.input_type
+class _FeatureEngineeringMaterializedFeatureState:
+    def __init__(__self__, *,
+                 feature_name: Optional[pulumi.Input[_builtins.str]] = None,
+                 last_materialization_time: Optional[pulumi.Input[_builtins.str]] = None,
+                 materialized_feature_id: Optional[pulumi.Input[_builtins.str]] = None,
+                 offline_store_config: Optional[pulumi.Input['FeatureEngineeringMaterializedFeatureOfflineStoreConfigArgs']] = None,
+                 online_store_config: Optional[pulumi.Input['FeatureEngineeringMaterializedFeatureOnlineStoreConfigArgs']] = None,
+                 pipeline_schedule_state: Optional[pulumi.Input[_builtins.str]] = None,
+                 table_name: Optional[pulumi.Input[_builtins.str]] = None):
+        """
+        Input properties used for looking up and filtering FeatureEngineeringMaterializedFeature resources.
+        :param pulumi.Input[_builtins.str] feature_name: The full name of the feature in Unity Catalog
+        :param pulumi.Input[_builtins.str] last_materialization_time: (string) - The timestamp when the pipeline last ran and updated the materialized feature values.
+               If the pipeline has not run yet, this field will be null
+        :param pulumi.Input[_builtins.str] materialized_feature_id: (string) - Unique identifier for the materialized feature
+        :param pulumi.Input[_builtins.str] pipeline_schedule_state: The schedule state of the materialization pipeline. Possible values are: `ACTIVE`, `PAUSED`, `SNAPSHOT`
+        :param pulumi.Input[_builtins.str] table_name: (string) - The fully qualified Unity Catalog path to the table containing the materialized feature (Delta table or Lakebase table). Output only
+        """
+        if feature_name is not None:
+            pulumi.set(__self__, "feature_name", feature_name)
+        if last_materialization_time is not None:
+            pulumi.set(__self__, "last_materialization_time", last_materialization_time)
+        if materialized_feature_id is not None:
+            pulumi.set(__self__, "materialized_feature_id", materialized_feature_id)
+        if offline_store_config is not None:
+            pulumi.set(__self__, "offline_store_config", offline_store_config)
+        if online_store_config is not None:
+            pulumi.set(__self__, "online_store_config", online_store_config)
+        if pipeline_schedule_state is not None:
+            pulumi.set(__self__, "pipeline_schedule_state", pipeline_schedule_state)
+        if table_name is not None:
+            pulumi.set(__self__, "table_name", table_name)
+
+    @_builtins.property
+    @pulumi.getter(name="featureName")
+    def feature_name(self) -> Optional[pulumi.Input[_builtins.str]]:
+        """
+        The full name of the feature in Unity Catalog
+        """
+        return pulumi.get(self, "feature_name")
+
+    @feature_name.setter
+    def feature_name(self, value: Optional[pulumi.Input[_builtins.str]]):
+        pulumi.set(self, "feature_name", value)
+
+    @_builtins.property
+    @pulumi.getter(name="lastMaterializationTime")
+    def last_materialization_time(self) -> Optional[pulumi.Input[_builtins.str]]:
+        """
+        (string) - The timestamp when the pipeline last ran and updated the materialized feature values.
+        If the pipeline has not run yet, this field will be null
+        """
+        return pulumi.get(self, "last_materialization_time")
+
+    @last_materialization_time.setter
+    def last_materialization_time(self, value: Optional[pulumi.Input[_builtins.str]]):
+        pulumi.set(self, "last_materialization_time", value)
+
+    @_builtins.property
+    @pulumi.getter(name="materializedFeatureId")
+    def materialized_feature_id(self) -> Optional[pulumi.Input[_builtins.str]]:
+        """
+        (string) - Unique identifier for the materialized feature
+        """
+        return pulumi.get(self, "materialized_feature_id")
+
+    @materialized_feature_id.setter
+    def materialized_feature_id(self, value: Optional[pulumi.Input[_builtins.str]]):
+        pulumi.set(self, "materialized_feature_id", value)
+
+    @_builtins.property
+    @pulumi.getter(name="offlineStoreConfig")
+    def offline_store_config(self) -> Optional[pulumi.Input['FeatureEngineeringMaterializedFeatureOfflineStoreConfigArgs']]:
+        return pulumi.get(self, "offline_store_config")
+
+    @offline_store_config.setter
+    def offline_store_config(self, value: Optional[pulumi.Input['FeatureEngineeringMaterializedFeatureOfflineStoreConfigArgs']]):
+        pulumi.set(self, "offline_store_config", value)
+
+    @_builtins.property
+    @pulumi.getter(name="onlineStoreConfig")
+    def online_store_config(self) -> Optional[pulumi.Input['FeatureEngineeringMaterializedFeatureOnlineStoreConfigArgs']]:
+        return pulumi.get(self, "online_store_config")
+
+    @online_store_config.setter
+    def online_store_config(self, value: Optional[pulumi.Input['FeatureEngineeringMaterializedFeatureOnlineStoreConfigArgs']]):
+        pulumi.set(self, "online_store_config", value)
+
+    @_builtins.property
+    @pulumi.getter(name="pipelineScheduleState")
+    def pipeline_schedule_state(self) -> Optional[pulumi.Input[_builtins.str]]:
+        """
+        The schedule state of the materialization pipeline. Possible values are: `ACTIVE`, `PAUSED`, `SNAPSHOT`
+        """
+        return pulumi.get(self, "pipeline_schedule_state")
+
+    @pipeline_schedule_state.setter
+    def pipeline_schedule_state(self, value: Optional[pulumi.Input[_builtins.str]]):
+        pulumi.set(self, "pipeline_schedule_state", value)
+
+    @_builtins.property
+    @pulumi.getter(name="tableName")
+    def table_name(self) -> Optional[pulumi.Input[_builtins.str]]:
+        """
+        (string) - The fully qualified Unity Catalog path to the table containing the materialized feature (Delta table or Lakebase table). Output only
+        """
+        return pulumi.get(self, "table_name")
+
+    @table_name.setter
+    def table_name(self, value: Optional[pulumi.Input[_builtins.str]]):
+        pulumi.set(self, "table_name", value)
+
+
+@pulumi.type_token("databricks:index/featureEngineeringMaterializedFeature:FeatureEngineeringMaterializedFeature")
+class FeatureEngineeringMaterializedFeature(pulumi.CustomResource):
+    @overload
+    def __init__(__self__,
+                 resource_name: str,
+                 opts: Optional[pulumi.ResourceOptions] = None,
+                 feature_name: Optional[pulumi.Input[_builtins.str]] = None,
+                 offline_store_config: Optional[pulumi.Input[Union['FeatureEngineeringMaterializedFeatureOfflineStoreConfigArgs', 'FeatureEngineeringMaterializedFeatureOfflineStoreConfigArgsDict']]] = None,
+                 online_store_config: Optional[pulumi.Input[Union['FeatureEngineeringMaterializedFeatureOnlineStoreConfigArgs', 'FeatureEngineeringMaterializedFeatureOnlineStoreConfigArgsDict']]] = None,
+                 pipeline_schedule_state: Optional[pulumi.Input[_builtins.str]] = None,
+                 __props__=None):
+        """
+        [![Private Preview](https://img.shields.io/badge/Release_Stage-Private_Preview-blueviolet)](https://docs.databricks.com/aws/en/release-notes/release-types)
+
+        ## Import
+
+        As of Pulumi v1.5, resources can be imported through configuration.
+
+        hcl
+
+        import {
+
+          id = "materialized_feature_id"
+
+          to = databricks_feature_engineering_materialized_feature.this
+
+        }
+
+        If you are using an older version of Pulumi, import the resource using the `pulumi import` command as follows:
+
+        ```sh
+        $ pulumi import databricks:index/featureEngineeringMaterializedFeature:FeatureEngineeringMaterializedFeature this "materialized_feature_id"
+        ```
+
+        :param str resource_name: The name of the resource.
+        :param pulumi.ResourceOptions opts: Options for the resource.
+        :param pulumi.Input[_builtins.str] feature_name: The full name of the feature in Unity Catalog
+        :param pulumi.Input[_builtins.str] pipeline_schedule_state: The schedule state of the materialization pipeline. Possible values are: `ACTIVE`, `PAUSED`, `SNAPSHOT`
+        """
+        ...
+    @overload
+    def __init__(__self__,
+                 resource_name: str,
+                 args: FeatureEngineeringMaterializedFeatureArgs,
+                 opts: Optional[pulumi.ResourceOptions] = None):
+        """
+        [![Private Preview](https://img.shields.io/badge/Release_Stage-Private_Preview-blueviolet)](https://docs.databricks.com/aws/en/release-notes/release-types)
+
+        ## Import
+
+        As of Pulumi v1.5, resources can be imported through configuration.
+
+        hcl
+
+        import {
+
+          id = "materialized_feature_id"
+
+          to = databricks_feature_engineering_materialized_feature.this
+
+        }
+
+        If you are using an older version of Pulumi, import the resource using the `pulumi import` command as follows:
+
+        ```sh
+        $ pulumi import databricks:index/featureEngineeringMaterializedFeature:FeatureEngineeringMaterializedFeature this "materialized_feature_id"
+        ```
+
+        :param str resource_name: The name of the resource.
+        :param FeatureEngineeringMaterializedFeatureArgs args: The arguments to use to populate this resource's properties.
+        :param pulumi.ResourceOptions opts: Options for the resource.
+        """
+        ...
+    def __init__(__self__, resource_name: str, *args, **kwargs):
+        resource_args, opts = _utilities.get_resource_args_opts(FeatureEngineeringMaterializedFeatureArgs, pulumi.ResourceOptions, *args, **kwargs)
+        if resource_args is not None:
+            __self__._internal_init(resource_name, opts, **resource_args.__dict__)
+        else:
+            __self__._internal_init(resource_name, *args, **kwargs)
+
+    def _internal_init(__self__,
+                 resource_name: str,
+                 opts: Optional[pulumi.ResourceOptions] = None,
+                 feature_name: Optional[pulumi.Input[_builtins.str]] = None,
+                 offline_store_config: Optional[pulumi.Input[Union['FeatureEngineeringMaterializedFeatureOfflineStoreConfigArgs', 'FeatureEngineeringMaterializedFeatureOfflineStoreConfigArgsDict']]] = None,
+                 online_store_config: Optional[pulumi.Input[Union['FeatureEngineeringMaterializedFeatureOnlineStoreConfigArgs', 'FeatureEngineeringMaterializedFeatureOnlineStoreConfigArgsDict']]] = None,
+                 pipeline_schedule_state: Optional[pulumi.Input[_builtins.str]] = None,
+                 __props__=None):
+        opts = pulumi.ResourceOptions.merge(_utilities.get_resource_opts_defaults(), opts)
+        if not isinstance(opts, pulumi.ResourceOptions):
+            raise TypeError('Expected resource options to be a ResourceOptions instance')
+        if opts.id is None:
+            if __props__ is not None:
+                raise TypeError('__props__ is only valid when passed in combination with a valid opts.id to get an existing resource')
+            __props__ = FeatureEngineeringMaterializedFeatureArgs.__new__(FeatureEngineeringMaterializedFeatureArgs)
+
+            if feature_name is None and not opts.urn:
+                raise TypeError("Missing required property 'feature_name'")
+            __props__.__dict__["feature_name"] = feature_name
+            __props__.__dict__["offline_store_config"] = offline_store_config
+            __props__.__dict__["online_store_config"] = online_store_config
+            __props__.__dict__["pipeline_schedule_state"] = pipeline_schedule_state
+            __props__.__dict__["last_materialization_time"] = None
+            __props__.__dict__["materialized_feature_id"] = None
+            __props__.__dict__["table_name"] = None
+        super(FeatureEngineeringMaterializedFeature, __self__).__init__(
+            'databricks:index/featureEngineeringMaterializedFeature:FeatureEngineeringMaterializedFeature',
+            resource_name,
+            __props__,
+            opts)
+
+    @staticmethod
+    def get(resource_name: str,
+            id: pulumi.Input[str],
+            opts: Optional[pulumi.ResourceOptions] = None,
+            feature_name: Optional[pulumi.Input[_builtins.str]] = None,
+            last_materialization_time: Optional[pulumi.Input[_builtins.str]] = None,
+            materialized_feature_id: Optional[pulumi.Input[_builtins.str]] = None,
+            offline_store_config: Optional[pulumi.Input[Union['FeatureEngineeringMaterializedFeatureOfflineStoreConfigArgs', 'FeatureEngineeringMaterializedFeatureOfflineStoreConfigArgsDict']]] = None,
+            online_store_config: Optional[pulumi.Input[Union['FeatureEngineeringMaterializedFeatureOnlineStoreConfigArgs', 'FeatureEngineeringMaterializedFeatureOnlineStoreConfigArgsDict']]] = None,
+            pipeline_schedule_state: Optional[pulumi.Input[_builtins.str]] = None,
+            table_name: Optional[pulumi.Input[_builtins.str]] = None) -> 'FeatureEngineeringMaterializedFeature':
+        """
+        Get an existing FeatureEngineeringMaterializedFeature resource's state with the given name, id, and optional extra
+        properties used to qualify the lookup.
+
+        :param str resource_name: The unique name of the resulting resource.
+        :param pulumi.Input[str] id: The unique provider ID of the resource to lookup.
+        :param pulumi.ResourceOptions opts: Options for the resource.
+        :param pulumi.Input[_builtins.str] feature_name: The full name of the feature in Unity Catalog
+        :param pulumi.Input[_builtins.str] last_materialization_time: (string) - The timestamp when the pipeline last ran and updated the materialized feature values.
+               If the pipeline has not run yet, this field will be null
+        :param pulumi.Input[_builtins.str] materialized_feature_id: (string) - Unique identifier for the materialized feature
+        :param pulumi.Input[_builtins.str] pipeline_schedule_state: The schedule state of the materialization pipeline. Possible values are: `ACTIVE`, `PAUSED`, `SNAPSHOT`
+        :param pulumi.Input[_builtins.str] table_name: (string) - The fully qualified Unity Catalog path to the table containing the materialized feature (Delta table or Lakebase table). Output only
+        """
+        opts = pulumi.ResourceOptions.merge(opts, pulumi.ResourceOptions(id=id))
+
+        __props__ = _FeatureEngineeringMaterializedFeatureState.__new__(_FeatureEngineeringMaterializedFeatureState)
+
+        __props__.__dict__["feature_name"] = feature_name
+        __props__.__dict__["last_materialization_time"] = last_materialization_time
+        __props__.__dict__["materialized_feature_id"] = materialized_feature_id
+        __props__.__dict__["offline_store_config"] = offline_store_config
+        __props__.__dict__["online_store_config"] = online_store_config
+        __props__.__dict__["pipeline_schedule_state"] = pipeline_schedule_state
+        __props__.__dict__["table_name"] = table_name
+        return FeatureEngineeringMaterializedFeature(resource_name, opts=opts, __props__=__props__)
+
+    @_builtins.property
+    @pulumi.getter(name="featureName")
+    def feature_name(self) -> pulumi.Output[_builtins.str]:
+        """
+        The full name of the feature in Unity Catalog
+        """
+        return pulumi.get(self, "feature_name")
+
+    @_builtins.property
+    @pulumi.getter(name="lastMaterializationTime")
+    def last_materialization_time(self) -> pulumi.Output[_builtins.str]:
+        """
+        (string) - The timestamp when the pipeline last ran and updated the materialized feature values.
+        If the pipeline has not run yet, this field will be null
+        """
+        return pulumi.get(self, "last_materialization_time")
+
+    @_builtins.property
+    @pulumi.getter(name="materializedFeatureId")
+    def materialized_feature_id(self) -> pulumi.Output[_builtins.str]:
+        """
+        (string) - Unique identifier for the materialized feature
+        """
+        return pulumi.get(self, "materialized_feature_id")
+
+    @_builtins.property
+    @pulumi.getter(name="offlineStoreConfig")
+    def offline_store_config(self) -> pulumi.Output[Optional['outputs.FeatureEngineeringMaterializedFeatureOfflineStoreConfig']]:
+        return pulumi.get(self, "offline_store_config")
+
+    @_builtins.property
+    @pulumi.getter(name="onlineStoreConfig")
+    def online_store_config(self) -> pulumi.Output[Optional['outputs.FeatureEngineeringMaterializedFeatureOnlineStoreConfig']]:
+        return pulumi.get(self, "online_store_config")
+
+    @_builtins.property
+    @pulumi.getter(name="pipelineScheduleState")
+    def pipeline_schedule_state(self) -> pulumi.Output[Optional[_builtins.str]]:
+        """
+        The schedule state of the materialization pipeline. Possible values are: `ACTIVE`, `PAUSED`, `SNAPSHOT`
+        """
+        return pulumi.get(self, "pipeline_schedule_state")
+
+    @_builtins.property
+    @pulumi.getter(name="tableName")
+    def table_name(self) -> pulumi.Output[_builtins.str]:
+        """
+        (string) - The fully qualified Unity Catalog path to the table containing the materialized feature (Delta table or Lakebase table). Output only
+        """
+        return pulumi.get(self, "table_name")
+

pulumi_databricks/file.py

@@ -270,7 +270,7 @@ class File(pulumi.CustomResource):
 
         init_script = databricks.File("init_script",
             content_base64=std.base64encode(input=\"\"\"#!/bin/bash
-        echo "Hello World"
+        echo \\"Hello World\\"
         \"\"\").result,
             path=f"{this['volumePath']}/fileName")
         ```
@@ -368,7 +368,7 @@ class File(pulumi.CustomResource):
 
         init_script = databricks.File("init_script",
             content_base64=std.base64encode(input=\"\"\"#!/bin/bash
-        echo "Hello World"
+        echo \\"Hello World\\"
         \"\"\").result,
             path=f"{this['volumePath']}/fileName")
         ```

pulumi_databricks/get_account_federation_policies.py

@@ -27,10 +27,13 @@ class GetAccountFederationPoliciesResult:
     """
     A collection of values returned by getAccountFederationPolicies.
     """
-    def __init__(__self__, id=None, policies=None):
+    def __init__(__self__, id=None, page_size=None, policies=None):
         if id and not isinstance(id, str):
             raise TypeError("Expected argument 'id' to be a str")
         pulumi.set(__self__, "id", id)
+        if page_size and not isinstance(page_size, int):
+            raise TypeError("Expected argument 'page_size' to be a int")
+        pulumi.set(__self__, "page_size", page_size)
         if policies and not isinstance(policies, list):
             raise TypeError("Expected argument 'policies' to be a list")
         pulumi.set(__self__, "policies", policies)
@@ -43,6 +46,11 @@ class GetAccountFederationPoliciesResult:
         """
         return pulumi.get(self, "id")
 
+    @_builtins.property
+    @pulumi.getter(name="pageSize")
+    def page_size(self) -> Optional[_builtins.int]:
+        return pulumi.get(self, "page_size")
+
     @_builtins.property
     @pulumi.getter
     def policies(self) -> Sequence['outputs.GetAccountFederationPoliciesPolicyResult']:
@@ -56,11 +64,15 @@ class AwaitableGetAccountFederationPoliciesResult(GetAccountFederationPoliciesRe
             yield self
         return GetAccountFederationPoliciesResult(
             id=self.id,
+            page_size=self.page_size,
             policies=self.policies)
 
 
-def get_account_federation_policies(opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetAccountFederationPoliciesResult:
+def get_account_federation_policies(page_size: Optional[_builtins.int] = None,
+                                    opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetAccountFederationPoliciesResult:
     """
+    [![GA](https://img.shields.io/badge/Release_Stage-GA-green)](https://docs.databricks.com/aws/en/release-notes/release-types)
+
     This data source can be used to fetch the list of account federation policies.
 
     > **Note** This data source can only be used with an account-level provider!
@@ -77,14 +89,19 @@ def get_account_federation_policies(opts: Optional[pulumi.InvokeOptions] = None)
     ```
     """
     __args__ = dict()
+    __args__['pageSize'] = page_size
     opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke('databricks:index/getAccountFederationPolicies:getAccountFederationPolicies', __args__, opts=opts, typ=GetAccountFederationPoliciesResult).value
 
     return AwaitableGetAccountFederationPoliciesResult(
         id=pulumi.get(__ret__, 'id'),
+        page_size=pulumi.get(__ret__, 'page_size'),
         policies=pulumi.get(__ret__, 'policies'))
-def get_account_federation_policies_output(opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetAccountFederationPoliciesResult]:
+def get_account_federation_policies_output(page_size: Optional[pulumi.Input[Optional[_builtins.int]]] = None,
+                                           opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetAccountFederationPoliciesResult]:
     """
+    [![GA](https://img.shields.io/badge/Release_Stage-GA-green)](https://docs.databricks.com/aws/en/release-notes/release-types)
+
     This data source can be used to fetch the list of account federation policies.
 
     > **Note** This data source can only be used with an account-level provider!
@@ -101,8 +118,10 @@ def get_account_federation_policies_output(opts: Optional[Union[pulumi.InvokeOpt
     ```
     """
     __args__ = dict()
+    __args__['pageSize'] = page_size
     opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke_output('databricks:index/getAccountFederationPolicies:getAccountFederationPolicies', __args__, opts=opts, typ=GetAccountFederationPoliciesResult)
     return __ret__.apply(lambda __response__: GetAccountFederationPoliciesResult(
         id=pulumi.get(__response__, 'id'),
+        page_size=pulumi.get(__response__, 'page_size'),
         policies=pulumi.get(__response__, 'policies')))

pulumi_databricks/get_account_federation_policy.py

@@ -14,7 +14,6 @@ else:
     from typing_extensions import NotRequired, TypedDict, TypeAlias
 from . import _utilities
 from . import outputs
-from ._inputs import *
 
 __all__ = [
     'GetAccountFederationPolicyResult',
@@ -67,7 +66,7 @@ class GetAccountFederationPolicyResult:
 
     @_builtins.property
     @pulumi.getter
-    def description(self) -> Optional[_builtins.str]:
+    def description(self) -> _builtins.str:
         """
         (string) - Description of the federation policy
         """
@@ -96,7 +95,7 @@ class GetAccountFederationPolicyResult:
 
     @_builtins.property
     @pulumi.getter(name="oidcPolicy")
-    def oidc_policy(self) -> Optional['outputs.GetAccountFederationPolicyOidcPolicyResult']:
+    def oidc_policy(self) -> 'outputs.GetAccountFederationPolicyOidcPolicyResult':
         """
         (OidcFederationPolicy)
         """
@@ -106,7 +105,7 @@ class GetAccountFederationPolicyResult:
     @pulumi.getter(name="policyId")
     def policy_id(self) -> _builtins.str:
         """
-        (string) - The ID of the federation policy
+        (string) - The ID of the federation policy. Output only
         """
         return pulumi.get(self, "policy_id")
 
@@ -114,7 +113,7 @@ class GetAccountFederationPolicyResult:
     @pulumi.getter(name="servicePrincipalId")
     def service_principal_id(self) -> _builtins.int:
         """
-        (integer) - The service principal ID that this federation policy applies to. Only set for service principal federation policies
+        (integer) - The service principal ID that this federation policy applies to. Output only. Only set for service principal federation policies
         """
         return pulumi.get(self, "service_principal_id")
 
@@ -152,12 +151,11 @@ class AwaitableGetAccountFederationPolicyResult(GetAccountFederationPolicyResult
             update_time=self.update_time)
 
 
-def get_account_federation_policy(description: Optional[_builtins.str] = None,
-                                  oidc_policy: Optional[Union['GetAccountFederationPolicyOidcPolicyArgs', 'GetAccountFederationPolicyOidcPolicyArgsDict']] = None,
-                                  policy_id: Optional[_builtins.str] = None,
-                                  service_principal_id: Optional[_builtins.int] = None,
+def get_account_federation_policy(policy_id: Optional[_builtins.str] = None,
                                   opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetAccountFederationPolicyResult:
     """
+    [![GA](https://img.shields.io/badge/Release_Stage-GA-green)](https://docs.databricks.com/aws/en/release-notes/release-types)
+
     This data source can be used to get a single account federation policy.
 
     > **Note** This data source can only be used with an account-level provider!
@@ -167,16 +165,10 @@ def get_account_federation_policy(description: Optional[_builtins.str] = None,
     Referring to an account federation policy by id:
 
 
-    :param _builtins.str description: (string) - Description of the federation policy
-    :param Union['GetAccountFederationPolicyOidcPolicyArgs', 'GetAccountFederationPolicyOidcPolicyArgsDict'] oidc_policy: (OidcFederationPolicy)
-    :param _builtins.str policy_id: The ID of the federation policy
-    :param _builtins.int service_principal_id: (integer) - The service principal ID that this federation policy applies to. Only set for service principal federation policies
+    :param _builtins.str policy_id: The ID of the federation policy. Output only
     """
     __args__ = dict()
-    __args__['description'] = description
-    __args__['oidcPolicy'] = oidc_policy
     __args__['policyId'] = policy_id
-    __args__['servicePrincipalId'] = service_principal_id
     opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke('databricks:index/getAccountFederationPolicy:getAccountFederationPolicy', __args__, opts=opts, typ=GetAccountFederationPolicyResult).value
 
@@ -190,12 +182,11 @@ def get_account_federation_policy(description: Optional[_builtins.str] = None,
         service_principal_id=pulumi.get(__ret__, 'service_principal_id'),
         uid=pulumi.get(__ret__, 'uid'),
         update_time=pulumi.get(__ret__, 'update_time'))
-def get_account_federation_policy_output(description: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
-                                         oidc_policy: Optional[pulumi.Input[Optional[Union['GetAccountFederationPolicyOidcPolicyArgs', 'GetAccountFederationPolicyOidcPolicyArgsDict']]]] = None,
-                                         policy_id: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
-                                         service_principal_id: Optional[pulumi.Input[Optional[_builtins.int]]] = None,
+def get_account_federation_policy_output(policy_id: Optional[pulumi.Input[_builtins.str]] = None,
                                          opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetAccountFederationPolicyResult]:
     """
+    [![GA](https://img.shields.io/badge/Release_Stage-GA-green)](https://docs.databricks.com/aws/en/release-notes/release-types)
+
     This data source can be used to get a single account federation policy.
 
     > **Note** This data source can only be used with an account-level provider!
@@ -205,16 +196,10 @@ def get_account_federation_policy_output(description: Optional[pulumi.Input[Opti
     Referring to an account federation policy by id:
 
 
-    :param _builtins.str description: (string) - Description of the federation policy
-    :param Union['GetAccountFederationPolicyOidcPolicyArgs', 'GetAccountFederationPolicyOidcPolicyArgsDict'] oidc_policy: (OidcFederationPolicy)
-    :param _builtins.str policy_id: The ID of the federation policy
-    :param _builtins.int service_principal_id: (integer) - The service principal ID that this federation policy applies to. Only set for service principal federation policies
+    :param _builtins.str policy_id: The ID of the federation policy. Output only
     """
     __args__ = dict()
-    __args__['description'] = description
-    __args__['oidcPolicy'] = oidc_policy
     __args__['policyId'] = policy_id
-    __args__['servicePrincipalId'] = service_principal_id
     opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke_output('databricks:index/getAccountFederationPolicy:getAccountFederationPolicy', __args__, opts=opts, typ=GetAccountFederationPolicyResult)
     return __ret__.apply(lambda __response__: GetAccountFederationPolicyResult(

pulumi_databricks/get_account_network_policies.py

@@ -61,6 +61,8 @@ class AwaitableGetAccountNetworkPoliciesResult(GetAccountNetworkPoliciesResult):
 
 def get_account_network_policies(opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetAccountNetworkPoliciesResult:
     """
+    [![GA](https://img.shields.io/badge/Release_Stage-GA-green)](https://docs.databricks.com/aws/en/release-notes/release-types)
+
     This data source can be used to fetch the list of network policies.
 
     > **Note** This data source can only be used with an account-level provider!
@@ -85,6 +87,8 @@ def get_account_network_policies(opts: Optional[pulumi.InvokeOptions] = None) ->
         items=pulumi.get(__ret__, 'items'))
 def get_account_network_policies_output(opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetAccountNetworkPoliciesResult]:
     """
+    [![GA](https://img.shields.io/badge/Release_Stage-GA-green)](https://docs.databricks.com/aws/en/release-notes/release-types)
+
     This data source can be used to fetch the list of network policies.
 
     > **Note** This data source can only be used with an account-level provider!