semantic-link-labs 0.7.0__tar.gz → 0.7.2__tar.gz

{semantic_link_labs-0.7.0 → semantic_link_labs-0.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: semantic-link-labs
-Version: 0.7.0
+Version: 0.7.2
 Summary: Semantic Link Labs for Microsoft Fabric
 Author: Microsoft Corporation
 License: MIT License
@@ -25,7 +25,7 @@ Requires-Dist: pytest>=8.2.1; extra == "test"
 # Semantic Link Labs
 
 [![PyPI version](https://badge.fury.io/py/semantic-link-labs.svg)](https://badge.fury.io/py/semantic-link-labs)
-[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.7.0&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
+[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.7.2&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![Downloads](https://static.pepy.tech/badge/semantic-link-labs)](https://pepy.tech/project/semantic-link-labs)
 

{semantic_link_labs-0.7.0 → semantic_link_labs-0.7.2}/README.md

@@ -1,7 +1,7 @@
 # Semantic Link Labs
 
 [![PyPI version](https://badge.fury.io/py/semantic-link-labs.svg)](https://badge.fury.io/py/semantic-link-labs)
-[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.7.0&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
+[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.7.2&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![Downloads](https://static.pepy.tech/badge/semantic-link-labs)](https://pepy.tech/project/semantic-link-labs)
 

{semantic_link_labs-0.7.0 → semantic_link_labs-0.7.2}/docs/requirements.txt

@@ -8,4 +8,5 @@ azure-keyvault-secrets
 azure-storage-file-datalake==12.3.1
 azure-storage-blob>=12.9.0
 anytree
-IPython
+IPython
+polib

{semantic_link_labs-0.7.0 → semantic_link_labs-0.7.2}/docs/source/conf.py

@@ -13,7 +13,7 @@ sys.path.insert(0, os.path.abspath('../../src/'))
 project = 'semantic-link-labs'
 copyright = '2024, Microsoft and community'
 author = 'Microsoft and community'
-release = '0.7.0'
+release = '0.7.2'
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

{semantic_link_labs-0.7.0 → semantic_link_labs-0.7.2}/pyproject.toml

@@ -7,7 +7,7 @@ name="semantic-link-labs"
 authors = [
     { name = "Microsoft Corporation" },
 ]
-version="0.7.0"
+version="0.7.2"
 description="Semantic Link Labs for Microsoft Fabric"
 readme="README.md"
 requires-python=">=3.10,<3.12"

{semantic_link_labs-0.7.0 → semantic_link_labs-0.7.2}/src/semantic_link_labs.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: semantic-link-labs
-Version: 0.7.0
+Version: 0.7.2
 Summary: Semantic Link Labs for Microsoft Fabric
 Author: Microsoft Corporation
 License: MIT License
@@ -25,7 +25,7 @@ Requires-Dist: pytest>=8.2.1; extra == "test"
 # Semantic Link Labs
 
 [![PyPI version](https://badge.fury.io/py/semantic-link-labs.svg)](https://badge.fury.io/py/semantic-link-labs)
-[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.7.0&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
+[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.7.2&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![Downloads](https://static.pepy.tech/badge/semantic-link-labs)](https://pepy.tech/project/semantic-link-labs)
 

{semantic_link_labs-0.7.0 → semantic_link_labs-0.7.2}/src/sempy_labs/_helper_functions.py

@@ -11,6 +11,7 @@ from typing import Optional, Tuple, List
 from uuid import UUID
 import sempy_labs._icons as icons
 from sempy.fabric.exceptions import FabricHTTPException
+import urllib.parse
 
 
 def create_abfss_path(
@@ -681,7 +682,8 @@ def resolve_workspace_capacity(workspace: Optional[str] = None) -> Tuple[UUID, s
     """
 
     workspace = fabric.resolve_workspace_name(workspace)
-    dfW = fabric.list_workspaces(filter=f"name eq '{workspace}'")
+    filter_condition = urllib.parse.quote(workspace)
+    dfW = fabric.list_workspaces(filter=f"name eq '{filter_condition}'")
     capacity_id = dfW["Capacity Id"].iloc[0]
     dfC = fabric.list_capacities()
     dfC_filt = dfC[dfC["Id"] == capacity_id]
@@ -711,7 +713,8 @@ def get_capacity_id(workspace: Optional[str] = None) -> UUID:
     """
 
     workspace = fabric.resolve_workspace_name(workspace)
-    dfW = fabric.list_workspaces(filter=f"name eq '{workspace}'")
+    filter_condition = urllib.parse.quote(workspace)
+    dfW = fabric.list_workspaces(filter=f"name eq '{filter_condition}'")
     if len(dfW) == 0:
         raise ValueError(f"{icons.red_dot} The '{workspace}' does not exist'.")
 

{semantic_link_labs-0.7.0 → semantic_link_labs-0.7.2}/src/sempy_labs/_list_functions.py

@@ -8,12 +8,11 @@ from sempy_labs._helper_functions import (
     _decode_b64,
     pagination,
     lro,
+    resolve_item_type,
 )
 import pandas as pd
 import base64
 import requests
-import time
-import json
 from pyspark.sql import SparkSession
 from typing import Optional
 import sempy_labs._icons as icons
@@ -1529,7 +1528,7 @@ def list_shortcuts(
     lakehouse: Optional[str] = None, workspace: Optional[str] = None
 ) -> pd.DataFrame:
     """
-    Shows all shortcuts which exist in a Fabric lakehouse.
+    Shows all shortcuts which exist in a Fabric lakehouse and their properties.
 
     Parameters
     ----------
@@ -1551,71 +1550,84 @@ def list_shortcuts(
 
     if lakehouse is None:
         lakehouse_id = fabric.get_lakehouse_id()
-        lakehouse = resolve_lakehouse_name(lakehouse_id, workspace)
     else:
         lakehouse_id = resolve_lakehouse_id(lakehouse, workspace)
 
+    client = fabric.FabricRestClient()
+
     df = pd.DataFrame(
         columns=[
             "Shortcut Name",
             "Shortcut Path",
-            "Source",
-            "Source Lakehouse Name",
+            "Source Type",
+            "Source Workspace Id",
             "Source Workspace Name",
-            "Source Path",
-            "Source Connection ID",
-            "Source Location",
-            "Source SubPath",
+            "Source Item Id",
+            "Source Item Name",
+            "Source Item Type",
+            "OneLake Path",
+            "Connection Id",
+            "Location",
+            "Bucket",
+            "SubPath",
         ]
     )
 
-    client = fabric.FabricRestClient()
     response = client.get(
         f"/v1/workspaces/{workspace_id}/items/{lakehouse_id}/shortcuts"
     )
+
     if response.status_code != 200:
         raise FabricHTTPException(response)
 
     responses = pagination(client, response)
 
     for r in responses:
-        for s in r.get("value", []):
-            shortcutName = s.get("name")
-            shortcutPath = s.get("path")
-            source = list(s["target"].keys())[0]
-            (
-                sourceLakehouseName,
-                sourceWorkspaceName,
-                sourcePath,
-                connectionId,
-                location,
-                subpath,
-            ) = (None, None, None, None, None, None)
-            if source == "oneLake":
-                sourceLakehouseId = s.get("target", {}).get(source, {}).get("itemId")
-                sourcePath = s.get("target", {}).get(source, {}).get("path")
-                sourceWorkspaceId = (
-                    s.get("target", {}).get(source, {}).get("workspaceId")
-                )
-                sourceWorkspaceName = fabric.resolve_workspace_name(sourceWorkspaceId)
-                sourceLakehouseName = resolve_lakehouse_name(
-                    sourceLakehouseId, sourceWorkspaceName
-                )
-            else:
-                connectionId = s.get("target", {}).get(source, {}).get("connectionId")
-                location = s.get("target", {}).get(source, {}).get("location")
-                subpath = s.get("target", {}).get(source, {}).get("subpath")
+        for i in r.get("value", []):
+            tgt = i.get("target", {})
+            s3_compat = tgt.get("s3Compatible", {})
+            gcs = tgt.get("googleCloudStorage", {})
+            eds = tgt.get("externalDataShare", {})
+            connection_id = (
+                s3_compat.get("connectionId")
+                or gcs.get("connectionId")
+                or eds.get("connectionId")
+                or None
+            )
+            location = s3_compat.get("location") or gcs.get("location") or None
+            sub_path = s3_compat.get("subpath") or gcs.get("subpath") or None
+            source_workspace_id = tgt.get("oneLake", {}).get("workspaceId")
+            source_item_id = tgt.get("oneLake", {}).get("itemId")
+            source_workspace_name = (
+                fabric.resolve_workspace_name(source_workspace_id)
+                if source_workspace_id is not None
+                else None
+            )
 
             new_data = {
-                "Shortcut Name": shortcutName,
-                "Shortcut Path": shortcutPath,
-                "Source": source,
-                "Source Lakehouse Name": sourceLakehouseName,
-                "Source Workspace Name": sourceWorkspaceName,
-                "Source Path": sourcePath,
-                "Source Connection ID": connectionId,
-                "Source Location": location,
-                "Source SubPath": subpath,
+                "Shortcut Name": i.get("name"),
+                "Shortcut Path": i.get("path"),
+                "Source Type": tgt.get("type"),
+                "Source Workspace Id": source_workspace_id,
+                "Source Workspace Name": source_workspace_name,
+                "Source Item Id": source_item_id,
+                "Source Item Name": (
+                    fabric.resolve_item_name(
+                        source_item_id, workspace=source_workspace_name
+                    )
+                    if source_item_id is not None
+                    else None
+                ),
+                "Source Item Type": (
+                    resolve_item_type(source_item_id, workspace=source_workspace_name)
+                    if source_item_id is not None
+                    else None
+                ),
+                "OneLake Path": tgt.get("oneLake", {}).get("path"),
+                "Connection Id": connection_id,
+                "Location": location,
+                "Bucket": s3_compat.get("bucket"),
+                "SubPath": sub_path,
             }
             df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
@@ -1722,9 +1734,9 @@ def create_custom_pool(
     min_node_count : int
         The `minimum node count <https://learn.microsoft.com/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool?tabs=HTTP#autoscaleproperties>`_.
     max_node_count : int
-        The `maximum node count <https://learn.microsoft.com/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool?tabs=HTTP#autoscaleproperties>`_.
+        The `maximum node count <https://learn.microsoft.com/en-us/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool?tabs=HTTP#autoscaleproperties>`_.
     min_executors : int
-        The `minimum executors <https://learn.microsoft.com/en-us/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool?tabs=HTTP#dynamicexecutorallocationproperties>`_.
+        The `minimum executors <https://learn.microsoft.com/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool?tabs=HTTP#dynamicexecutorallocationproperties>`_.
     max_executors : int
         The `maximum executors <https://learn.microsoft.com/en-us/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool?tabs=HTTP#dynamicexecutorallocationproperties>`_.
     node_family : str, default='MemoryOptimized'
@@ -1799,10 +1811,10 @@ def update_custom_pool(
         The `minimum node count <https://learn.microsoft.com/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool?tabs=HTTP#autoscaleproperties>`_.
         Defaults to None which keeps the existing property setting.
     max_node_count : int, default=None
-        The `maximum node count <https://learn.microsoft.com/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool?tabs=HTTP#autoscaleproperties>`_.
+        The `maximum node count <https://learn.microsoft.com/en-us/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool?tabs=HTTP#autoscaleproperties>`_.
         Defaults to None which keeps the existing property setting.
     min_executors : int, default=None
-        The `minimum executors <https://learn.microsoft.com/en-us/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool?tabs=HTTP#dynamicexecutorallocationproperties>`_.
+        The `minimum executors <https://learn.microsoft.com/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool?tabs=HTTP#dynamicexecutorallocationproperties>`_.
         Defaults to None which keeps the existing property setting.
     max_executors : int, default=None
         The `maximum executors <https://learn.microsoft.com/en-us/rest/api/fabric/spark/custom-pools/create-workspace-custom-pool?tabs=HTTP#dynamicexecutorallocationproperties>`_.
@@ -2092,7 +2104,7 @@ def update_spark_settings(
         `Default pool <https://learn.microsoft.com/rest/api/fabric/spark/workspace-settings/update-spark-settings?tabs=HTTP#poolproperties>`_ for workspace.
         Defaults to None which keeps the existing property setting.
     max_node_count : int, default=None
-        The `maximum node count <https://learn.microsoft.com/rest/api/fabric/spark/workspace-settings/update-spark-settings?tabs=HTTP#starterpoolproperties>`_.
+        The `maximum node count <https://learn.microsoft.com/en-us/rest/api/fabric/spark/workspace-settings/update-spark-settings?tabs=HTTP#starterpoolproperties>`_.
         Defaults to None which keeps the existing property setting.
     max_executors : int, default=None
         The `maximum executors <https://learn.microsoft.com/rest/api/fabric/spark/workspace-settings/update-spark-settings?tabs=HTTP#starterpoolproperties>`_.
@@ -2161,7 +2173,10 @@ def update_spark_settings(
 
 
 def add_user_to_workspace(
-    email_address: str, role_name: str, workspace: Optional[str] = None
+    email_address: str,
+    role_name: str,
+    principal_type: Optional[str] = "User",
+    workspace: Optional[str] = None,
 ):
     """
     Adds a user to a workspace.
@@ -2172,13 +2187,12 @@ def add_user_to_workspace(
         The email address of the user.
     role_name : str
         The `role <https://learn.microsoft.com/rest/api/power-bi/groups/add-group-user#groupuseraccessright>`_ of the user within the workspace.
+    principal_type : str, default='User'
+        The `principal type <https://learn.microsoft.com/rest/api/power-bi/groups/add-group-user#principaltype>`_.
     workspace : str, default=None
         The name of the workspace.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
-
-    Returns
-    -------
     """
 
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
@@ -2190,10 +2204,21 @@ def add_user_to_workspace(
             f"{icons.red_dot} Invalid role. The 'role_name' parameter must be one of the following: {role_names}."
         )
     plural = "n" if role_name == "Admin" else ""
+    principal_types = ["App", "Group", "None", "User"]
+    principal_type = principal_type.capitalize()
+    if principal_type not in principal_types:
+        raise ValueError(
+            f"{icons.red_dot} Invalid princpal type. Valid options: {principal_types}."
+        )
 
     client = fabric.PowerBIRestClient()
 
-    request_body = {"emailAddress": email_address, "groupUserAccessRight": role_name}
+    request_body = {
+        "emailAddress": email_address,
+        "groupUserAccessRight": role_name,
+        "principalType": principal_type,
+        "identifier": email_address,
+    }
 
     response = client.post(
         f"/v1.0/myorg/groups/{workspace_id}/users", json=request_body
@@ -2236,7 +2261,10 @@ def delete_user_from_workspace(email_address: str, workspace: Optional[str] = No
 
 
 def update_workspace_user(
-    email_address: str, role_name: str, workspace: Optional[str] = None
+    email_address: str,
+    role_name: str,
+    principal_type: Optional[str] = "User",
+    workspace: Optional[str] = None,
 ):
     """
     Updates a user's role within a workspace.
@@ -2247,13 +2275,12 @@ def update_workspace_user(
         The email address of the user.
     role_name : str
         The `role <https://learn.microsoft.com/rest/api/power-bi/groups/add-group-user#groupuseraccessright>`_ of the user within the workspace.
+    principal_type : str, default='User'
+        The `principal type <https://learn.microsoft.com/rest/api/power-bi/groups/add-group-user#principaltype>`_.
     workspace : str, default=None
         The name of the workspace.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
-
-    Returns
-    -------
     """
 
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
@@ -2264,8 +2291,19 @@ def update_workspace_user(
         raise ValueError(
             f"{icons.red_dot} Invalid role. The 'role_name' parameter must be one of the following: {role_names}."
         )
+    principal_types = ["App", "Group", "None", "User"]
+    principal_type = principal_type.capitalize()
+    if principal_type not in principal_types:
+        raise ValueError(
+            f"{icons.red_dot} Invalid princpal type. Valid options: {principal_types}."
+        )
 
-    request_body = {"emailAddress": email_address, "groupUserAccessRight": role_name}
+    request_body = {
+        "emailAddress": email_address,
+        "groupUserAccessRight": role_name,
+        "principalType": principal_type,
+        "identifier": email_address,
+    }
 
     client = fabric.PowerBIRestClient()
     response = client.put(f"/v1.0/myorg/groups/{workspace_id}/users", json=request_body)

{semantic_link_labs-0.7.0 → semantic_link_labs-0.7.2}/src/sempy_labs/_model_bpa_bulk.py

@@ -21,6 +21,7 @@ def run_model_bpa_bulk(
     extended: Optional[bool] = False,
     language: Optional[str] = None,
     workspace: Optional[str | List[str]] = None,
+    skip_models: Optional[str | List[str]] = ["ModelBPA", "Fabric Capacity Metrics"],
 ):
     """
     Runs the semantic model Best Practice Analyzer across all semantic models in a workspace (or all accessible workspaces).
@@ -41,18 +42,22 @@ def run_model_bpa_bulk(
     workspace : str | List[str], default=None
         The workspace or list of workspaces to scan.
         Defaults to None which scans all accessible workspaces.
-
-    Returns
-    -------
+    skip_models : str | List[str], default=['ModelBPA', 'Fabric Capacity Metrics']
+        The semantic models to always skip when running this analysis.
     """
 
     import pyspark.sql.functions as F
 
     if not lakehouse_attached():
         raise ValueError(
-            "No lakehouse is attached to this notebook. Must attach a lakehouse to the notebook."
+            f"{icons.red_dot} No lakehouse is attached to this notebook. Must attach a lakehouse to the notebook."
         )
 
+    if isinstance(skip_models, str):
+        skip_models = [skip_models]
+
+    skip_models.extend(["ModelBPA", "Fabric Capacity Metrics"])
+
     cols = [
         "Capacity Name",
         "Capacity Id",
@@ -113,8 +118,7 @@ def run_model_bpa_bulk(
                 or set(["Lakehouse", "SemanticModel"]).issubset(set(x["Type"]))
             )
             default_semantic_models = filtered_df["Display Name"].unique().tolist()
-            # Skip ModelBPA :)
-            skip_models = default_semantic_models + [icons.model_bpa_name]
+            skip_models.extend(default_semantic_models)
             dfD_filt = dfD[~dfD["Dataset Name"].isin(skip_models)]
 
             if len(dfD_filt) > 0:

{semantic_link_labs-0.7.0 → semantic_link_labs-0.7.2}/src/sempy_labs/_model_bpa_rules.py

@@ -135,6 +135,17 @@ def model_bpa_rules(
                     "Setting the 'Data Coverage Definition' property may lead to better performance because the engine knows when it can only query the import-portion of the table and when it needs to query the DirectQuery portion of the table.",
                     "https://learn.microsoft.com/analysis-services/tom/table-partitions?view=asallproducts-allversions",
                 ),
+                (
+                    "Performance",
+                    "Model",
+                    "Warning",
+                    "Dual mode is only relevant for dimension tables if DirectQuery is used for the corresponding fact table",
+                    lambda obj: not any(
+                        p.Mode == TOM.ModeType.DirectQuery for p in tom.all_partitions()
+                    )
+                    and any(p.Mode == TOM.ModeType.Dual for p in tom.all_partitions()),
+                    "Only use Dual mode for dimension tables/partitions where a corresponding fact table is in DirectQuery. Using Dual mode in other circumstances (i.e. rest of the model is in Import mode) may lead to performance issues especially if the number of measures in the model is high.",
+                ),
                 (
                     "Performance",
                     "Table",
@@ -590,13 +601,13 @@ def model_bpa_rules(
                         re.search(
                             r"USERELATIONSHIP\s*\(\s*\'*"
                             + obj.FromTable.Name
-                            + "'*\["
+                            + r"'*\["
                             + obj.FromColumn.Name
-                            + "\]\s*,\s*'*"
+                            + r"\]\s*,\s*'*"
                             + obj.ToTable.Name
-                            + "'*\["
+                            + r"'*\["
                             + obj.ToColumn.Name
-                            + "\]",
+                            + r"\]",
                             m.Expression,
                             flags=re.IGNORECASE,
                         )

{semantic_link_labs-0.7.0 → semantic_link_labs-0.7.2}/src/sempy_labs/_query_scale_out.py

@@ -21,10 +21,6 @@ def qso_sync(dataset: str, workspace: Optional[str] = None):
         The Fabric workspace name.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
-
-    Returns
-    -------
-
     """
 
     # https://learn.microsoft.com/en-us/rest/api/power-bi/datasets/trigger-query-scale-out-sync-in-group
@@ -63,7 +59,6 @@ def qso_sync_status(
     -------
     Tuple[pandas.DataFrame, pandas.DataFrame]
         2 pandas dataframes showing the query scale-out sync status.
-
     """
 
     # https://learn.microsoft.com/en-us/rest/api/power-bi/datasets/get-query-scale-out-sync-status-in-group
@@ -161,7 +156,6 @@ def disable_qso(dataset: str, workspace: Optional[str] = None) -> pd.DataFrame:
     -------
     pandas.DataFrame
         A pandas dataframe showing the current query scale out settings.
-
     """
 
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
@@ -177,6 +171,7 @@ def disable_qso(dataset: str, workspace: Optional[str] = None) -> pd.DataFrame:
         raise FabricHTTPException(response)
 
     df = list_qso_settings(dataset=dataset, workspace=workspace)
+
     print(
         f"{icons.green_dot} Query scale out has been disabled for the '{dataset}' semantic model within the '{workspace}' workspace."
     )
@@ -210,14 +205,20 @@ def set_qso(
     -------
     pandas.DataFrame
         A pandas dataframe showing the current query scale-out settings.
-
     """
 
+    from sempy_labs._helper_functions import is_default_semantic_model
+
     # https://learn.microsoft.com/en-us/rest/api/power-bi/datasets/update-dataset-in-group
 
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
     dataset_id = resolve_dataset_id(dataset, workspace)
 
+    if is_default_semantic_model(dataset=dataset, workspace=workspace):
+        raise ValueError(
+            f"{icons.red_dot} The 'set_qso' function does not run against default semantic models."
+        )
+
     if max_read_only_replicas == 0:
         disable_qso(dataset=dataset, workspace=workspace)
         return
@@ -225,32 +226,33 @@ def set_qso(
     request_body = {
         "queryScaleOutSettings": {
             "autoSyncReadOnlyReplicas": auto_sync,
-            "maxReadOnlyReplicas": str(max_read_only_replicas),
+            "maxReadOnlyReplicas": max_read_only_replicas,
         }
     }
 
-    ssm = set_semantic_model_storage_format(
-        dataset=dataset, storage_format="Large", workspace=workspace
-    )
-    if ssm == 200:
-        client = fabric.PowerBIRestClient()
-        response = client.patch(
-            f"/v1.0/myorg/groups/{workspace_id}/datasets/{dataset_id}",
-            json=request_body,
-        )
-        if response.status_code != 200:
-            raise FabricHTTPException(response)
+    dfL = list_qso_settings(dataset=dataset, workspace=workspace)
+    storage_mode = dfL["Storage Mode"].iloc[0]
 
-        df = list_qso_settings(dataset=dataset, workspace=workspace)
-        print(
-            f"{icons.green_dot} Query scale out has been set on the '{dataset}' semantic model within the '{workspace}' workspace."
-        )
-        return df
-    else:
-        raise ValueError(
-            f"{icons.red_dot} Failed to set the '{dataset}' semantic model within the '{workspace}' workspace to large semantic model storage format. This is a prerequisite for enabling Query Scale Out.\n\"https://learn.microsoft.com/power-bi/enterprise/service-premium-scale-out#prerequisites\""
+    if storage_mode == "Small":
+        set_semantic_model_storage_format(
+            dataset=dataset, storage_format="Large", workspace=workspace
         )
 
+    client = fabric.PowerBIRestClient()
+    response = client.patch(
+        f"/v1.0/myorg/groups/{workspace_id}/datasets/{dataset_id}",
+        json=request_body,
+    )
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    df = list_qso_settings(dataset=dataset, workspace=workspace)
+    print(
+        f"{icons.green_dot} Query scale out has been set on the '{dataset}' semantic model within the '{workspace}' workspace."
+    )
+
+    return df
+
 
 def set_semantic_model_storage_format(
     dataset: str, storage_format: str, workspace: Optional[str] = None
@@ -268,10 +270,6 @@ def set_semantic_model_storage_format(
         The Fabric workspace name.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
-
-    Returns
-    -------
-
     """
 
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)

{semantic_link_labs-0.7.0 → semantic_link_labs-0.7.2}/src/sempy_labs/lakehouse/__init__.py

@@ -3,6 +3,7 @@ from sempy_labs.lakehouse._get_lakehouse_tables import get_lakehouse_tables
 from sempy_labs.lakehouse._lakehouse import (
     lakehouse_attached,
     optimize_lakehouse_tables,
+    vacuum_lakehouse_tables,
 )
 
 from sempy_labs.lakehouse._shortcuts import (
@@ -19,4 +20,5 @@ __all__ = [
     # create_shortcut,
     "create_shortcut_onelake",
     "delete_shortcut",
+    "vacuum_lakehouse_tables",
 ]