semantic-link-labs 0.4.1__py3-none-any.whl → 0.4.2__py3-none-any.whl

sempy_labs/directlake/_show_unsupported_directlake_objects.py

@@ -4,8 +4,9 @@ import pandas as pd
 from sempy_labs._list_functions import list_tables
 from sempy_labs._helper_functions import format_dax_object_name
 from typing import Optional, Tuple
+from sempy._utils._log import log
 
-
+@log
 def show_unsupported_direct_lake_objects(
     dataset: str, workspace: Optional[str] = None
 ) -> Tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame]:
@@ -29,9 +30,8 @@ def show_unsupported_direct_lake_objects(
 
     pd.options.mode.chained_assignment = None
 
-    if workspace == None:
-        workspace_id = fabric.get_workspace_id()
-        workspace = fabric.resolve_workspace_name(workspace_id)
+    if workspace is None:
+        workspace = fabric.resolve_workspace_name()
 
     dfT = list_tables(dataset, workspace)
     dfC = fabric.list_columns(dataset=dataset, workspace=workspace)

sempy_labs/directlake/_update_directlake_model_lakehouse_connection.py

@@ -5,8 +5,9 @@ from sempy_labs._helper_functions import (
     resolve_lakehouse_name,
     resolve_workspace_name_and_id,
 )
-from sempy_labs._tom import connect_semantic_model
-from typing import List, Optional, Union
+from sempy_labs.tom import connect_semantic_model
+from typing import Optional
+import sempy_labs._icons as icons
 
 
 def update_direct_lake_model_lakehouse_connection(
@@ -41,10 +42,10 @@ def update_direct_lake_model_lakehouse_connection(
 
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
-    if lakehouse_workspace == None:
+    if lakehouse_workspace is None:
         lakehouse_workspace = workspace
 
-    if lakehouse == None:
+    if lakehouse is None:
         lakehouse_id = fabric.get_lakehouse_id()
         lakehouse = resolve_lakehouse_name(lakehouse_id, lakehouse_workspace)
 
@@ -54,7 +55,7 @@ def update_direct_lake_model_lakehouse_connection(
 
     if len(dfI_filt) == 0:
         print(
-            f"The '{lakehouse}' lakehouse does not exist within the '{lakehouse_workspace}' workspace. Therefore it cannot be used to support the '{dataset}' semantic model within the '{workspace}' workspace."
+            f"{icons.red_dot} The '{lakehouse}' lakehouse does not exist within the '{lakehouse_workspace}' workspace. Therefore it cannot be used to support the '{dataset}' semantic model within the '{workspace}' workspace."
         )
 
     dfP = fabric.list_partitions(dataset=dataset, workspace=workspace)
@@ -62,7 +63,7 @@ def update_direct_lake_model_lakehouse_connection(
 
     if len(dfP_filt) == 0:
         print(
-            f"The '{dataset}' semantic model is not in Direct Lake. This function is only applicable to Direct Lake semantic models."
+            f"{icons.yellow_dot} The '{dataset}' semantic model is not in Direct Lake. This function is only applicable to Direct Lake semantic models."
         )
     else:
         with connect_semantic_model(
@@ -73,9 +74,7 @@ def update_direct_lake_model_lakehouse_connection(
             try:
                 tom.model.Expressions["DatabaseQuery"].Expression = shEx
                 print(
-                    f"The expression in the '{dataset}' semantic model has been updated to point to the '{lakehouse}' lakehouse in the '{lakehouse_workspace}' workspace."
-                )
-            except:
-                print(
-                    f"ERROR: The expression in the '{dataset}' semantic model was not updated."
+                    f"{icons.green_dot} The expression in the '{dataset}' semantic model has been updated to point to the '{lakehouse}' lakehouse in the '{lakehouse_workspace}' workspace."
                 )
+            except Exception as e:
+                raise ValueError(f"{icons.red_dot} The expression in the '{dataset}' semantic model was not updated.") from e

sempy_labs/directlake/_update_directlake_partition_entity.py

@@ -1,13 +1,17 @@
+import sempy
 import sempy.fabric as fabric
-from sempy_labs._tom import connect_semantic_model
+from sempy_labs.tom import connect_semantic_model
+from sempy_labs._helper_functions import resolve_lakehouse_name
 from typing import List, Optional, Union
-
+import sempy_labs._icons as icons
 
 def update_direct_lake_partition_entity(
     dataset: str,
     table_name: Union[str, List[str]],
     entity_name: Union[str, List[str]],
     workspace: Optional[str] = None,
+    lakehouse: Optional[str] = None,
+    lakehouse_workspace: Optional[str] = None
 ):
     """
     Remaps a table (or tables) in a Direct Lake semantic model to a table in a lakehouse.
@@ -24,10 +28,24 @@ def update_direct_lake_partition_entity(
         The Fabric workspace name in which the semantic model exists.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
+    lakehouse : str, default=None
+        The Fabric lakehouse used by the Direct Lake semantic model.
+        Defaults to None which resolves to the lakehouse attached to the notebook.
+    lakehouse_workspace : str, default=None
+        The Fabric workspace used by the lakehouse.
+        Defaults to None which resolves to the workspace of the attached lakehouse
+        or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
     workspace = fabric.resolve_workspace_name(workspace)
 
+    if lakehouse_workspace is None:
+        lakehouse_workspace = workspace
+
+    if lakehouse is None:
+        lakehouse_id = fabric.get_lakehouse_id()
+        lakehouse = resolve_lakehouse_name(lakehouse_id, lakehouse_workspace)
+
     # Support both str & list types
     if isinstance(table_name, str):
         table_name = [table_name]
@@ -36,7 +54,7 @@ def update_direct_lake_partition_entity(
 
     if len(table_name) != len(entity_name):
         print(
-            f"ERROR: The 'table_name' and 'entity_name' arrays must be of equal length."
+            f"{icons.red_dot} The 'table_name' and 'entity_name' arrays must be of equal length."
         )
         return
 
@@ -46,7 +64,7 @@ def update_direct_lake_partition_entity(
 
         if not tom.is_direct_lake():
             print(
-                f"The '{dataset}' semantic model within the '{workspace}' workspace is not in Direct Lake mode."
+                f"{icons.yellow_dot} The '{dataset}' semantic model within the '{workspace}' workspace is not in Direct Lake mode."
             )
             return
 
@@ -56,9 +74,7 @@ def update_direct_lake_partition_entity(
             try:
                 tom.model.Tables[tName].Partitions[0].EntityName = eName
                 print(
-                    f"The '{tName}' table in the '{dataset}' semantic model has been updated to point to the '{eName}' table in the '{lakehouse}' lakehouse within the '{lakehouse_workspace}' workspace."
-                )
-            except:
-                print(
-                    f"ERROR: The '{tName}' table in the '{dataset}' semantic model has not been updated."
+                    f"{icons.green_dot} The '{tName}' table in the '{dataset}' semantic model has been updated to point to the '{eName}' table in the '{lakehouse}' lakehouse within the '{lakehouse_workspace}' workspace."
                 )
+            except Exception as e:
+                raise ValueError(f"{icons.red_dot} The '{tName}' table in the '{dataset}' semantic model has not been updated.") from e

sempy_labs/directlake/_warm_cache.py

@@ -131,7 +131,7 @@ def warm_direct_lake_cache_perspective(
         bar.set_description(f"Warming the '{tableName}' table...")
         css = ",".join(map(str, filtered_list))
         dax = """EVALUATE TOPN(1,SUMMARIZECOLUMNS(""" + css + "))" ""
-        x = fabric.evaluate_dax(dataset=dataset, dax_string=dax, workspace=workspace)
+        fabric.evaluate_dax(dataset=dataset, dax_string=dax, workspace=workspace)
 
     print(f"{icons.green_dot} The following columns have been put into memory:")
 
@@ -168,10 +168,9 @@ def warm_direct_lake_cache_isresident(
 
     dfP = fabric.list_partitions(dataset=dataset, workspace=workspace)
     if not any(r["Mode"] == "DirectLake" for i, r in dfP.iterrows()):
-        print(
-            f"The '{dataset}' semantic model in the '{workspace}' workspace is not in Direct Lake mode. This function is specifically for semantic models in Direct Lake mode."
+        raise ValueError(
+            f"{icons.red_dot} The '{dataset}' semantic model in the '{workspace}' workspace is not in Direct Lake mode. This function is specifically for semantic models in Direct Lake mode."
         )
-        return
 
     # Identify columns which are currently in memory (Is Resident = True)
     dfC = fabric.list_columns(dataset=dataset, workspace=workspace, extended=True)
@@ -181,10 +180,9 @@ def warm_direct_lake_cache_isresident(
     dfC_filtered = dfC[dfC["Is Resident"]]
 
     if len(dfC_filtered) == 0:
-        print(
+        raise ValueError(
             f"{icons.yellow_dot} At present, no columns are in memory in the '{dataset}' semantic model in the '{workspace}' workspace."
         )
-        return
 
     # Refresh/frame dataset
     refresh_semantic_model(dataset=dataset, refresh_type="full", workspace=workspace)
@@ -199,7 +197,7 @@ def warm_direct_lake_cache_isresident(
         bar.set_description(f"Warming the '{tableName}' table...")
         css = ",".join(map(str, column_values))
         dax = """EVALUATE TOPN(1,SUMMARIZECOLUMNS(""" + css + "))" ""
-        x = fabric.evaluate_dax(dataset=dataset, dax_string=dax, workspace=workspace)
+        fabric.evaluate_dax(dataset=dataset, dax_string=dax, workspace=workspace)
 
     print(
         f"{icons.green_dot} The following columns have been put into memory. Temperature indicates the column temperature prior to the semantic model refresh."

sempy_labs/lakehouse/__init__.py

@@ -6,7 +6,6 @@ from sempy_labs.lakehouse._lakehouse import (
 )
 
 from sempy_labs.lakehouse._shortcuts import (
-    list_shortcuts,
     # create_shortcut,
     create_shortcut_onelake,
     delete_shortcut,
@@ -17,7 +16,6 @@ __all__ = [
     "get_lakehouse_tables",
     "lakehouse_attached",
     "optimize_lakehouse_tables",
-    "list_shortcuts",
     # create_shortcut,
     "create_shortcut_onelake",
     "delete_shortcut",

sempy_labs/lakehouse/_get_lakehouse_columns.py

@@ -8,8 +8,9 @@ from sempy_labs._helper_functions import (
     resolve_lakehouse_id,
 )
 from typing import Optional
+from sempy._utils._log import log
 
-
+@log
 def get_lakehouse_columns(
     lakehouse: Optional[str] = None, workspace: Optional[str] = None
 ):
@@ -47,7 +48,7 @@ def get_lakehouse_columns(
 
     workspace = fabric.resolve_workspace_name(workspace)
 
-    if lakehouse == None:
+    if lakehouse is None:
         lakehouse_id = fabric.get_lakehouse_id()
         lakehouse = resolve_lakehouse_name(lakehouse_id, workspace)
     else:

sempy_labs/lakehouse/_get_lakehouse_tables.py

@@ -1,3 +1,4 @@
+import sempy
 import sempy.fabric as fabric
 import pandas as pd
 from pyspark.sql import SparkSession
@@ -14,8 +15,10 @@ from sempy_labs.directlake._guardrails import (
 )
 from sempy_labs.lakehouse._lakehouse import lakehouse_attached
 from typing import Optional
+import sempy_labs._icons as icons
+from sempy._utils._log import log
 
-
+@log
 def get_lakehouse_tables(
     lakehouse: Optional[str] = None,
     workspace: Optional[str] = None,
@@ -61,7 +64,7 @@ def get_lakehouse_tables(
 
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
-    if lakehouse == None:
+    if lakehouse is None:
         lakehouse_id = fabric.get_lakehouse_id()
         lakehouse = resolve_lakehouse_name(lakehouse_id, workspace)
     else:
@@ -80,7 +83,7 @@ def get_lakehouse_tables(
         tType = i["type"]
         tFormat = i["format"]
         tLocation = i["location"]
-        if extended == False:
+        if not extended:
             new_data = {
                 "Workspace Name": workspace,
                 "Lakehouse Name": lakehouse,
@@ -170,9 +173,9 @@ def get_lakehouse_tables(
 
     if export:
         lakeAttach = lakehouse_attached()
-        if lakeAttach == False:
+        if lakeAttach is False:
             print(
-                f"In order to save the report.json file, a lakehouse must be attached to the notebook. Please attach a lakehouse to this notebook."
+                f"{icons.red_dot} In order to save the report.json file, a lakehouse must be attached to the notebook. Please attach a lakehouse to this notebook."
             )
             return
         spark = SparkSession.builder.getOrCreate()
@@ -234,7 +237,7 @@ def get_lakehouse_tables(
                     export_df[c] = export_df[c].astype(bool)
 
         print(
-            f"Saving Lakehouse table properties to the '{lakeTName}' table in the lakehouse...\n"
+            f"{icons.in_progress} Saving Lakehouse table properties to the '{lakeTName}' table in the lakehouse...\n"
         )
         now = datetime.datetime.now()
         export_df["Timestamp"] = now
@@ -244,7 +247,7 @@ def get_lakehouse_tables(
         spark_df = spark.createDataFrame(export_df)
         spark_df.write.mode("append").format("delta").saveAsTable(lakeTName)
         print(
-            f"\u2022 Lakehouse table properties have been saved to the '{lakeTName}' delta table."
+            f"{icons.bullet} Lakehouse table properties have been saved to the '{lakeTName}' delta table."
         )
 
     return df

sempy_labs/lakehouse/_lakehouse.py

@@ -4,7 +4,8 @@ from tqdm.auto import tqdm
 from pyspark.sql import SparkSession
 from sempy_labs._helper_functions import resolve_lakehouse_name
 from typing import List, Optional, Union
-
+import sempy_labs._icons as icons
+from sempy._utils._log import log
 
 def lakehouse_attached() -> bool:
     """
@@ -24,7 +25,7 @@ def lakehouse_attached() -> bool:
     else:
         return False
 
-
+@log
 def optimize_lakehouse_tables(
     tables: Optional[Union[str, List[str]]] = None,
     lakehouse: Optional[str] = None,
@@ -49,11 +50,11 @@ def optimize_lakehouse_tables(
     from sempy_labs.lakehouse._get_lakehouse_tables import get_lakehouse_tables
     from delta import DeltaTable
 
-    if workspace == None:
+    if workspace is None:
         workspace_id = fabric.get_workspace_id()
         workspace = fabric.resolve_workspace_name(workspace_id)
 
-    if lakehouse == None:
+    if lakehouse is None:
         lakehouse_id = fabric.get_lakehouse_id()
         lakehouse = resolve_lakehouse_name(lakehouse_id, workspace)
 
@@ -80,6 +81,6 @@ def optimize_lakehouse_tables(
         deltaTable = DeltaTable.forPath(spark, tablePath)
         deltaTable.optimize().executeCompaction()
         print(
-            f"The '{tableName}' table has been optimized. ({str(i)}/{str(tableCount)})"
+            f"{icons.green_dot} The '{tableName}' table has been optimized. ({str(i)}/{str(tableCount)})"
         )
         i += 1

sempy_labs/lakehouse/_shortcuts.py

@@ -1,12 +1,11 @@
 import sempy
 import sempy.fabric as fabric
-import pandas as pd
 from sempy_labs._helper_functions import (
     resolve_lakehouse_name,
     resolve_lakehouse_id,
     resolve_workspace_name_and_id,
 )
-from typing import List, Optional, Union
+from typing import Optional
 import sempy_labs._icons as icons
 
 
@@ -42,7 +41,7 @@ def create_shortcut_onelake(
     sourceWorkspaceId = fabric.resolve_workspace_id(source_workspace)
     sourceLakehouseId = resolve_lakehouse_id(source_lakehouse, source_workspace)
 
-    if destination_workspace == None:
+    if destination_workspace is None:
         destination_workspace = source_workspace
 
     destinationWorkspaceId = fabric.resolve_workspace_id(destination_workspace)
@@ -50,7 +49,7 @@ def create_shortcut_onelake(
         destination_lakehouse, destination_workspace
     )
 
-    if shortcut_name == None:
+    if shortcut_name is None:
         shortcut_name = table_name
 
     client = fabric.FabricRestClient()
@@ -80,9 +79,7 @@ def create_shortcut_onelake(
         else:
             print(response.status_code)
     except Exception as e:
-        print(
-            f"{icons.red_dot} Failed to create a shortcut for the '{table_name}' table: {e}"
-        )
+        raise ValueError(f"{icons.red_dot} Failed to create a shortcut for the '{table_name}' table.") from e
 
 
 def create_shortcut(
@@ -126,7 +123,7 @@ def create_shortcut(
 
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
-    if lakehouse == None:
+    if lakehouse is None:
         lakehouse_id = fabric.get_lakehouse_id()
     else:
         lakehouse_id = resolve_lakehouse_id(lakehouse, workspace)
@@ -157,103 +154,8 @@ def create_shortcut(
             )
         else:
             print(response.status_code)
-    except:
-        print(
-            f"{icons.red_dot} Failed to create a shortcut for the '{shortcut_name}' table."
-        )
-
-
-def list_shortcuts(
-    lakehouse: Optional[str] = None, workspace: Optional[str] = None
-) -> pd.DataFrame:
-    """
-    Shows all shortcuts which exist in a Fabric lakehouse.
-
-    Parameters
-    ----------
-    lakehouse : str, default=None
-        The Fabric lakehouse name.
-        Defaults to None which resolves to the lakehouse attached to the notebook.
-    workspace : str, default=None
-        The name of the Fabric workspace in which lakehouse resides.
-        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
-    -------
-    pandas.DataFrame
-        A pandas dataframe showing all the shortcuts which exist in the specified lakehouse.
-    """
-
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
-
-    if lakehouse == None:
-        lakehouse_id = fabric.get_lakehouse_id()
-        lakehouse = resolve_lakehouse_name(lakehouse_id, workspace)
-    else:
-        lakehouse_id = resolve_lakehouse_id(lakehouse, workspace)
-
-    df = pd.DataFrame(
-        columns=[
-            "Shortcut Name",
-            "Shortcut Path",
-            "Source",
-            "Source Lakehouse Name",
-            "Source Workspace Name",
-            "Source Path",
-            "Source Connection ID",
-            "Source Location",
-            "Source SubPath",
-        ]
-    )
-
-    client = fabric.FabricRestClient()
-    response = client.get(
-        f"/v1/workspaces/{workspace_id}/items/{lakehouse_id}/shortcuts"
-    )
-    if response.status_code == 200:
-        for s in response.json()["value"]:
-            shortcutName = s["name"]
-            shortcutPath = s["path"]
-            source = list(s["target"].keys())[0]
-            (
-                sourceLakehouseName,
-                sourceWorkspaceName,
-                sourcePath,
-                connectionId,
-                location,
-                subpath,
-            ) = (None, None, None, None, None, None)
-            if source == "oneLake":
-                sourceLakehouseId = s["target"][source]["itemId"]
-                sourcePath = s["target"][source]["path"]
-                sourceWorkspaceId = s["target"][source]["workspaceId"]
-                sourceWorkspaceName = fabric.resolve_workspace_name(sourceWorkspaceId)
-                sourceLakehouseName = resolve_lakehouse_name(
-                    sourceLakehouseId, sourceWorkspaceName
-                )
-            else:
-                connectionId = s["target"][source]["connectionId"]
-                location = s["target"][source]["location"]
-                subpath = s["target"][source]["subpath"]
-
-            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,
-            }
-            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
-
-    print(
-        f"This function relies on an API which is not yet official as of May 21, 2024. Once the API becomes official this function will work as expected."
-    )
-    return df
+    except Exception as e:
+        raise ValueError(f"{icons.red_dot} Failed to create a shortcut for the '{shortcut_name}' table.") from e
 
 
 def delete_shortcut(
@@ -277,7 +179,7 @@ def delete_shortcut(
 
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
-    if lakehouse == None:
+    if lakehouse is None:
         lakehouse_id = fabric.get_lakehouse_id()
         lakehouse = resolve_lakehouse_name(lakehouse_id, workspace)
     else:

sempy_labs/migration/__init__.py

@@ -14,7 +14,9 @@ from sempy_labs.migration._migrate_tables_columns_to_semantic_model import (
 )
 from sempy_labs.migration._migration_validation import (
     migration_validation,
-    # list_semantic_model_objects
+)
+from sempy_labs.migration._refresh_calc_tables import (
+    refresh_calc_tables,
 )
 
 __all__ = [
@@ -25,5 +27,5 @@ __all__ = [
     "migrate_model_objects_to_semantic_model",
     "migrate_tables_columns_to_semantic_model",
     "migration_validation",
-    # list_semantic_model_objects
+    "refresh_calc_tables"
 ]

sempy_labs/migration/_create_pqt_file.py

@@ -34,13 +34,13 @@ def create_pqt_file(
 
     lakeAttach = lakehouse_attached()
 
-    if lakeAttach == False:
+    if lakeAttach is False:
         print(
             f"{icons.red_dot} In order to run the 'create_pqt_file' function, a lakehouse must be attached to the notebook. Please attach a lakehouse to this notebook."
         )
         return
 
-    if workspace == None:
+    if workspace is None:
         workspace_id = fabric.get_workspace_id()
         workspace = fabric.resolve_workspace_name(workspace_id)
 

sempy_labs/migration/_migrate_calctables_to_lakehouse.py

@@ -8,7 +8,7 @@ from sempy_labs._helper_functions import (
     resolve_lakehouse_id,
     create_abfss_path,
 )
-from sempy_labs._tom import connect_semantic_model
+from sempy_labs.tom import connect_semantic_model
 from pyspark.sql import SparkSession
 from typing import List, Optional, Union
 from sempy._utils._log import log
@@ -52,16 +52,16 @@ def migrate_calc_tables_to_lakehouse(
 
     workspace = fabric.resolve_workspace_name(workspace)
 
-    if new_dataset_workspace == None:
+    if new_dataset_workspace is None:
         new_dataset_workspace = workspace
 
-    if lakehouse_workspace == None:
+    if lakehouse_workspace is None:
         lakehouse_workspace = new_dataset_workspace
         lakehouse_workspace_id = fabric.resolve_workspace_id(lakehouse_workspace)
     else:
         lakehouse_workspace_id = fabric.resolve_workspace_id(lakehouse_workspace)
 
-    if lakehouse == None:
+    if lakehouse is None:
         lakehouse_id = fabric.get_lakehouse_id()
         lakehouse = resolve_lakehouse_name(lakehouse_id, lakehouse_workspace)
     else:
@@ -288,16 +288,16 @@ def migrate_field_parameters(
         or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
-    from .HelperFunctions import format_dax_object_name
+    from sempy_labs import format_dax_object_name
 
     sempy.fabric._client._utils._init_analysis_services()
     import Microsoft.AnalysisServices.Tabular as TOM
 
-    if workspace == None:
+    if workspace is None:
         workspace_id = fabric.get_workspace_id()
         workspace = fabric.resolve_workspace_name(workspace_id)
 
-    if new_dataset_workspace == None:
+    if new_dataset_workspace is None:
         new_dataset_workspace = workspace
 
     dfC = fabric.list_columns(dataset=dataset, workspace=workspace)

sempy_labs/migration/_migrate_calctables_to_semantic_model.py

@@ -3,7 +3,7 @@ import sempy.fabric as fabric
 import re, datetime, time
 from sempy_labs.lakehouse._get_lakehouse_tables import get_lakehouse_tables
 from sempy_labs._helper_functions import resolve_lakehouse_name
-from sempy_labs._tom import connect_semantic_model
+from sempy_labs.tom import connect_semantic_model
 from typing import Optional
 from sempy._utils._log import log
 import sempy_labs._icons as icons
@@ -46,12 +46,12 @@ def migrate_calc_tables_to_semantic_model(
 
     workspace = fabric.resolve_workspace_name(workspace)
 
-    if new_dataset_workspace == None:
+    if new_dataset_workspace is None:
         new_dataset_workspace = workspace
 
-    if lakehouse_workspace == None:
+    if lakehouse_workspace is None:
         lakehouse_workspace = new_dataset_workspace
-    if lakehouse == None:
+    if lakehouse is None:
         lakehouse_id = fabric.get_lakehouse_id()
         lakehouse = resolve_lakehouse_name(lakehouse_id, lakehouse_workspace)
 

sempy_labs/migration/_migrate_model_objects_to_semantic_model.py

@@ -3,12 +3,11 @@ import sempy.fabric as fabric
 import re, datetime, time
 from sempy_labs._list_functions import list_tables
 from sempy_labs._helper_functions import create_relationship_name
-from sempy_labs._tom import connect_semantic_model
+from sempy_labs.tom import connect_semantic_model
 from typing import Optional
 from sempy._utils._log import log
 import sempy_labs._icons as icons
 
-
 @log
 def migrate_model_objects_to_semantic_model(
     dataset: str,
@@ -39,13 +38,13 @@ def migrate_model_objects_to_semantic_model(
     import Microsoft.AnalysisServices.Tabular as TOM
     import System
 
-    if workspace == None:
+    if workspace is None:
         workspace_id = fabric.get_workspace_id()
         workspace = fabric.resolve_workspace_name(workspace_id)
     else:
-        workspaceId = fabric.resolve_workspace_id(workspace)
+        workspace_id = fabric.resolve_workspace_id(workspace)
 
-    if new_dataset_workspace == None:
+    if new_dataset_workspace is None:
         new_dataset_workspace = workspace
 
     dfT = list_tables(dataset, workspace)
@@ -239,7 +238,7 @@ def migrate_model_objects_to_semantic_model(
                             f"\n{icons.in_progress} Updating calculation group column name..."
                         )
                         dfC_filt = dfC[
-                            (dfC["Table Name"] == cgName) & (dfC["Hidden"] == False)
+                            (dfC["Table Name"] == cgName) & (dfC["Hidden"] is False)
                         ]
                         colName = dfC_filt["Column Name"].iloc[0]
                         tom.model.Tables[cgName].Columns["Name"].Name = colName