Merge branch 'm-kovalsky/typing-format'

m-kovalsky · m-kovalsky · commit ae642b27ff91 · 2026-01-26T09:40:38.000+02:00
diff --git a/src/sempy_labs/_authentication.py b/src/sempy_labs/_authentication.py
@@ -121,7 +121,7 @@ def __call__(
         """
         Parameters
         ----------
-        audience : Literal["pbi", "storage", "azure", "graph", "asazure", "keyvault"] = "pbi") -> str
+        audience : typing.Literal["pbi", "storage", "azure", "graph", "asazure", "keyvault"] = "pbi") -> str
             Literal if it's for PBI/Fabric API call or OneLake/Storage Account call.
         region : str, default=None
             The region of the Azure Analysis Services. For example: 'westus2'.
diff --git a/src/sempy_labs/_dax.py b/src/sempy_labs/_dax.py
@@ -318,7 +318,7 @@ def _dax_perf_test(
 
     Returns
     -------
-    Tuple[pandas.DataFrame, dict]
+    typing.Tuple[pandas.DataFrame, dict]
         A pandas dataframe showing the SQL profiler trace results of the DAX queries.
         A dictionary of the query results in pandas dataframes.
     """
diff --git a/src/sempy_labs/_get_connection_string.py b/src/sempy_labs/_get_connection_string.py
@@ -24,7 +24,7 @@ def get_connection_string(
     ----------
     item : str | uuid.UUID
         The name or ID of the item (Lakehouse or MirroredDatabase).
-    type : Literal['Lakehouse', 'Warehouse', 'SQLEndpoint']
+    type : typing.Literal['Lakehouse', 'Warehouse', 'SQLEndpoint']
         The type of the item. Must be 'Lakehouse' or 'MirroredDatabase'.
     workspace : str | uuid.UUID, default=None
         The Fabric workspace name or ID.
diff --git a/src/sempy_labs/_helper_functions.py b/src/sempy_labs/_helper_functions.py
@@ -1126,7 +1126,7 @@ def resolve_workspace_name_and_id(
 
     Returns
     -------
-    str, uuid.UUID
+    typing.Tuple[str, str]
         The name and ID of the Fabric workspace.
     """
 
@@ -1349,7 +1349,7 @@ def resolve_dataset_from_report(
 
     Returns
     -------
-    Tuple[uuid.UUID, str, uuid.UUID, str]
+    typing.Tuple[uuid.UUID, str, uuid.UUID, str]
         The semantic model UUID, semantic model name, semantic model workspace UUID, semantic model workspace name
     """
 
@@ -1389,7 +1389,7 @@ def resolve_workspace_capacity(
 
     Returns
     -------
-    Tuple[uuid.UUID, str]
+    typing.Tuple[uuid.UUID, str]
         capacity Id; capacity came.
     """
     from sempy_labs._capacities import list_capacities
diff --git a/src/sempy_labs/_onelake.py b/src/sempy_labs/_onelake.py
@@ -149,7 +149,7 @@ def modify_immutability_policy(
     ----------
     retention_days : int
         Retention Days for the action.
-    scope : Literal["DiagnosticLogs"], default="DiagnosticLogs"
+    scope : typing.Literal["DiagnosticLogs"], default="DiagnosticLogs"
         The scope of the immutability policy. Currently, only "DiagnosticLogs" is supported.
     workspace : str | uuid.UUID, default=None
         The name or ID of the workspace.
diff --git a/src/sempy_labs/_query_scale_out.py b/src/sempy_labs/_query_scale_out.py
@@ -63,7 +63,7 @@ def qso_sync_status(
 
     Returns
     -------
-    Tuple[pandas.DataFrame, pandas.DataFrame]
+    typing.Tuple[pandas.DataFrame, pandas.DataFrame]
         2 pandas dataframes showing the query scale-out sync status.
     """
 
diff --git a/src/sempy_labs/_sql_audit_settings.py b/src/sempy_labs/_sql_audit_settings.py
@@ -51,7 +51,7 @@ def get_sql_audit_settings(
     ----------
     item : str | uuid.UUID
         The name or ID of the item (Warehouse or SQLEndpoint).
-    type : Literal['Warehouse', 'SQLEndpoint']
+    type : typing.Literal['Warehouse', 'SQLEndpoint']
         The type of the item. Must be 'Warehouse' or 'SQLEndpoint'.
     workspace : str | uuid.UUID, default=None
         The Fabric workspace name or ID.
@@ -113,7 +113,7 @@ def update_sql_audit_settings(
     ----------
     item : str | uuid.UUID
         The name or ID of the item (Warehouse or SQLEndpoint).
-    type : Literal['Warehouse', 'SQLEndpoint']
+    type : typing.Literal['Warehouse', 'SQLEndpoint']
         The type of the item. Must be 'Warehouse' or 'SQLEndpoint'.
     workspace : str | uuid.UUID, default=None
         The Fabric workspace name or ID.
@@ -175,7 +175,7 @@ def set_audit_actions_and_group(
     ----------
     item : str | uuid.UUID
         The name or ID of the item (Warehouse or SQLEndpoint).
-    type : Literal['Warehouse', 'SQLEndpoint']
+    type : typing.Literal['Warehouse', 'SQLEndpoint']
         The type of the item. Must be 'Warehouse' or 'SQLEndpoint'.
     sql_audit_groups : List[str]
         Set audit actions and groups request payload.
diff --git a/src/sempy_labs/_sql_endpoints.py b/src/sempy_labs/_sql_endpoints.py
@@ -80,13 +80,13 @@ def refresh_sql_endpoint_metadata(
     ----------
     item : str | uuid.UUID
         The name or ID of the item (Lakehouse or MirroredDatabase).
-    type : Literal['Lakehouse', 'MirroredDatabase']
+    type : typing.Literal['Lakehouse', 'MirroredDatabase']
         The type of the item. Must be 'Lakehouse' or 'MirroredDatabase'.
     workspace : str | uuid.UUID, default=None
         The Fabric workspace name or ID.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
-    timeout_unit : Literal['Seconds', 'Minutes', 'Hours', 'Days'], default='Minutes'
+    timeout_unit : typing.Literal['Seconds', 'Minutes', 'Hours', 'Days'], default='Minutes'
         The unit of time for the request duration before timing out. Additional duration types may be added over time.
     timeout_value : int, default=15
         The number of time units in the request duration.
diff --git a/src/sempy_labs/_takeover.py b/src/sempy_labs/_takeover.py
@@ -22,7 +22,7 @@ def takeover_item_ownership(
     ----------
     item : str | uuid.UUID
         The name or UUID of the Report or Semantic Model to take over.
-    type : Literal["Report", "SemanticModel"]
+    type : typing.Literal["Report", "SemanticModel"]
         The type of item to take over. Must be either 'Report' or 'SemanticModel'.
     workspace : Optional[str | uuid.UUID], default=None
         The name or ID of the workspace.
diff --git a/src/sempy_labs/_workspaces.py b/src/sempy_labs/_workspaces.py
@@ -260,9 +260,9 @@ def set_workspace_network_communication_policy(
 
     Parameters
     ----------
-    inbound_policy : Literal['Allow', 'Deny']
+    inbound_policy : typing.Literal['Allow', 'Deny']
         The policy for all inbound communications to a workspace.
-    outbound_policy : Literal['Allow', 'Deny']
+    outbound_policy : typing.Literal['Allow', 'Deny']
         The policy for all outbound communications to a workspace.
     workspace : str | uuid.UUID, default=None
         The workspace name or ID.
@@ -317,7 +317,7 @@ def set_workspace_git_outbound_policy(
 
     Parameters
     ----------
-    policy : Literal['Allow', 'Deny']
+    policy : typing.Literal['Allow', 'Deny']
         The policy for all Git outbound communications from a workspace.
     workspace : str | uuid.UUID, default=None
         The workspace name or ID.
diff --git a/src/sempy_labs/admin/_domains.py b/src/sempy_labs/admin/_domains.py
@@ -562,7 +562,7 @@ def sync_role_assignments_to_subdomains(
     ----------
     domain : str | uuid.UUID
         The domain name or ID.
-    role : Literal["Admin", "Contributor"]
+    role : typing.Literal["Admin", "Contributor"]
         The role to sync. Valid options: 'Admin', 'Contributor'.
     """
 
diff --git a/src/sempy_labs/admin/_labels.py b/src/sempy_labs/admin/_labels.py
@@ -69,7 +69,7 @@ def bulk_set_labels(
 
     label_id : uuid.UUID
         The label ID, which must be in the user's label policy.
-    assignment_method : Literal["Standard", "Priviledged"], default="Standard"
+    assignment_method : typing.Literal["Standard", "Priviledged"], default="Standard"
         Specifies whether the assigned label was set by an automated process or manually. Additional tenant setting property types may be added over time.
     """
 
diff --git a/src/sempy_labs/directlake/_dl_helper.py b/src/sempy_labs/directlake/_dl_helper.py
@@ -214,7 +214,7 @@ def get_direct_lake_source(
 
     Returns
     -------
-    Tuple[str, str, UUID, UUID]
+    typing.Tuple[str, str, uuid.UUID, uuid.UUID]
         If the source of the direct lake semantic model is a lakehouse this will return: 'Lakehouse', Lakehouse Name, SQL Endpoint Id, Workspace Id
         If the source of the direct lake semantic model is a warehouse this will return: 'Warehouse', Warehouse Name, Warehouse Id, Workspace Id
         If the semantic model is not a Direct Lake semantic model, it will return None, None, None.
diff --git a/src/sempy_labs/directlake/_get_directlake_lakehouse.py b/src/sempy_labs/directlake/_get_directlake_lakehouse.py
@@ -36,7 +36,7 @@ def get_direct_lake_lakehouse(
 
     Returns
     -------
-    Tuple[str, uuid.UUID]
+    typing.Tuple[str, uuid.UUID]
         The lakehouse name and lakehouse ID.
     """
 
diff --git a/src/sempy_labs/directlake/_show_unsupported_directlake_objects.py b/src/sempy_labs/directlake/_show_unsupported_directlake_objects.py
@@ -29,7 +29,7 @@ def show_unsupported_direct_lake_objects(
 
     Returns
     -------
-    pandas.DataFrame, pandas.DataFrame, pandas.DataFrame
+    typing.Tuple[pandas.DataFrame, pandas.DataFrame, pandas.DataFrame]
         3 pandas dataframes showing objects in a semantic model which are not supported by Direct Lake.
     """
 
diff --git a/src/sempy_labs/graph/_users.py b/src/sempy_labs/graph/_users.py
@@ -207,13 +207,13 @@ def send_mail(
         The email address of the recipients.
     content : str
         The email content.
-    content_type : Literal["Text", "HTML"], default="Text"
+    content_type : typing.Literal["Text", "HTML"], default="Text"
         The email content type. Options: "Text" or "HTML".
     cc_recipients : str | List[str], default=None
         The email address of the CC recipients.
     bcc_recipients : str | List[str], default=None
         The email address of the BCC recipients.
-    priority : Literal["Normal", "High", "Low"], default="Normal"
+    priority : typing.Literal["Normal", "High", "Low"], default="Normal"
         The email priority.
     attachments : str | List[str], default=None
         The abfss path or a list of the abfss paths of the attachments to include in the email.
diff --git a/src/sempy_labs/lakehouse/_helper.py b/src/sempy_labs/lakehouse/_helper.py
@@ -195,7 +195,7 @@ def load_table(
         The name of the table to load.
     file_path : str
         The path to the data to load.
-    mode : Literal["Overwrite", "Append"]
+    mode : typing.Literal["Overwrite", "Append"]
         The mode to use when loading the data.
         "Overwrite" will overwrite the existing data.
         "Append" will append the data to the existing data.
diff --git a/src/sempy_labs/report/_endorsement.py b/src/sempy_labs/report/_endorsement.py
@@ -24,7 +24,7 @@ def set_endorsement(
     ----------
     report : str | uuid.UUID
         Name or ID of the Power BI report.
-    endorsement : Literal["None", "Promoted", "Certified", "Master data"]
+    endorsement : typing.Literal["None", "Promoted", "Certified", "Master data"]
         The endorsement status to set for the report.
     workspace : str | uuid.UUID, default=None
         The workspace name or ID.
diff --git a/src/sempy_labs/report/_reportwrapper.py b/src/sempy_labs/report/_reportwrapper.py
@@ -508,7 +508,7 @@ def _resolve_page_name_and_display_name(
 
         Returns
         -------
-        Tuple[str, str]
+        typing.Tuple[str, str]
             The page name and display name.
         """
 
@@ -2557,7 +2557,7 @@ def _set_wallpaper_image(
             If None, applies to all pages.
         transparency : int, default=0
             The transparency level of the wallpaper image. Valid values are between 0 and 100.
-        image_fit : str, default="Normal"
+        image_fit : typing.Literal["Normal", "Fit", "Fill"], default="Normal"
             The fit type of the wallpaper image. Valid options: "Normal", "Fit", "Fill".
         """
         self._ensure_pbir()
diff --git a/src/sempy_labs/semantic_model/_copilot.py b/src/sempy_labs/semantic_model/_copilot.py
@@ -71,7 +71,7 @@ def set_endorsement(
     ----------
     dataset : str | uuid.UUID
         Name or ID of the semantic model.
-    endorsement : Literal["None", "Promoted", "Certified", "Master data"]
+    endorsement : typing.Literal["None", "Promoted", "Certified", "Master data"]
         The endorsement status to set for the semantic model.
     workspace : str | uuid.UUID, default=None
         The workspace name or ID.
diff --git a/src/sempy_labs/tom/_model.py b/src/sempy_labs/tom/_model.py
@@ -883,7 +883,7 @@ def set_ols(
             Name of the table.
         column_name : str, default=None
             Name of the column. Defaults to None which sets object level security for the entire table.
-        permission : Literal["Default", "None", "Read"], default="Default"
+        permission : typing.Literal["Default", "None", "Read"], default="Default"
             The object level security permission for the column.
             `Permission valid values <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.metadatapermission?view=analysisservices-dotnet>`_
         """
diff --git a/src/sempy_labs/variable_library/_functions.py b/src/sempy_labs/variable_library/_functions.py
@@ -688,7 +688,7 @@ def update_variable(
         Name of the variable.
     new_name : str, default=None
         New name of the variable. Defaults to None which keeps the same name.
-    type : Literal["Boolean", "DateTime", "Number", "Integer", "String"], default=None
+    type : typing.Literal["Boolean", "DateTime", "Number", "Integer", "String"], default=None
         New type of the variable. Valid types are: "Boolean", "DateTime", "Number", "Integer", "String". Defaults to None which keeps the same type.
     value : str, default=None
         New value of the variable. Defaults to None which keeps the same value.
diff --git a/src/sempy_labs/workspace/_items.py b/src/sempy_labs/workspace/_items.py
@@ -460,9 +460,9 @@ def set_network_communication_policy(
 
     Parameters
     ----------
-    inbound_policy : Literal['Allow', 'Deny']
+    inbound_policy : typing.Literal['Allow', 'Deny']
         The policy for all inbound communications to a workspace.
-    outbound_policy : Literal['Allow', 'Deny']
+    outbound_policy : typing.Literal['Allow', 'Deny']
         The policy for all outbound communications to a workspace.
     workspace : str | uuid.UUID, default=None
         The workspace name or ID.
@@ -554,7 +554,7 @@ def set_git_outbound_policy(
 
     Parameters
     ----------
-    policy : Literal['Allow', 'Deny']
+    policy : typing.Literal['Allow', 'Deny']
         The policy for all Git outbound communications from a workspace.
     workspace : str | uuid.UUID, default=None
         The workspace name or ID.
@@ -675,7 +675,7 @@ def set_outbound_gateway_rules(
 
     Parameters
     ----------
-    default_action : Literal['Allow', 'Deny']
+    default_action : typing.Literal['Allow', 'Deny']
         The default action for outbound gateway communication policy.
     allowed_gateways : List[uuid.UUID]
         A list of allowed gateway IDs.
