semantic-link-labs 0.8.10__py3-none-any.whl → 0.9.0__py3-none-any.whl

sempy_labs/_managed_private_endpoints.py

@@ -16,7 +16,7 @@ def create_managed_private_endpoint(
     target_private_link_resource_id: UUID,
     target_subresource_type: str,
     request_message: Optional[str] = None,
-    workspace: Optional[str] = None,
+    workspace: Optional[str | UUID] = None,
 ):
     """
     Creates a managed private endpoint.
@@ -27,19 +27,19 @@ def create_managed_private_endpoint(
     ----------
     name: str
         Name of the managed private endpoint.
-    target_private_link_resource_id: UUID
+    target_private_link_resource_id: uuid.UUID
         Resource Id of data source for which private endpoint needs to be created.
     target_subresource_type : str
         Sub-resource pointing to Private-link resoure.
     request_message : str, default=None
         Message to approve private endpoint request. Should not be more than 140 characters.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     request_body = {
         "name": name,
@@ -62,11 +62,13 @@ def create_managed_private_endpoint(
     lro(client, response, status_codes=[201, 202])
 
     print(
-        f"{icons.green_dot} The '{name}' managed private endpoint has been created within the '{workspace}' workspace."
+        f"{icons.green_dot} The '{name}' managed private endpoint has been created within the '{workspace_name}' workspace."
     )
 
 
-def list_managed_private_endpoints(workspace: Optional[str] = None) -> pd.DataFrame:
+def list_managed_private_endpoints(
+    workspace: Optional[str | UUID] = None,
+) -> pd.DataFrame:
     """
     Shows the managed private endpoints within a workspace.
 
@@ -74,8 +76,8 @@ def list_managed_private_endpoints(workspace: Optional[str] = None) -> pd.DataFr
 
     Parameters
     ----------
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -97,7 +99,7 @@ def list_managed_private_endpoints(workspace: Optional[str] = None) -> pd.DataFr
         ]
     )
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
     response = client.get(f"/v1/workspaces/{workspace_id}/managedPrivateEndpoints")
@@ -124,7 +126,7 @@ def list_managed_private_endpoints(workspace: Optional[str] = None) -> pd.DataFr
 
 
 def delete_managed_private_endpoint(
-    managed_private_endpoint: str, workspace: Optional[str] = None
+    managed_private_endpoint: str, workspace: Optional[str | UUID] = None
 ):
     """
     Deletes a Fabric managed private endpoint.
@@ -135,20 +137,20 @@ def delete_managed_private_endpoint(
     ----------
     managed_private_endpoint: str
         Name of the managed private endpoint.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
-    df = list_managed_private_endpoints(workspace=workspace)
+    df = list_managed_private_endpoints(workspace=workspace_id)
     df_filt = df[df["Managed Private Endpoint Name"] == managed_private_endpoint]
 
     if len(df_filt) == 0:
         raise ValueError(
-            f"{icons.red_dot} The '{managed_private_endpoint}' managed private endpoint does not exist within the '{workspace}' workspace."
+            f"{icons.red_dot} The '{managed_private_endpoint}' managed private endpoint does not exist within the '{workspace_name}' workspace."
         )
 
     item_id = df_filt["Managed Private Endpoint Id"].iloc[0]
@@ -162,5 +164,5 @@ def delete_managed_private_endpoint(
         raise FabricHTTPException(response)
 
     print(
-        f"{icons.green_dot} The '{managed_private_endpoint}' managed private endpoint within the '{workspace}' workspace has been deleted."
+        f"{icons.green_dot} The '{managed_private_endpoint}' managed private endpoint within the '{workspace_name}' workspace has been deleted."
     )

sempy_labs/_mirrored_databases.py

@@ -10,9 +10,10 @@ from sempy_labs._helper_functions import (
 from sempy.fabric.exceptions import FabricHTTPException
 import sempy_labs._icons as icons
 import base64
+from uuid import UUID
 
 
-def list_mirrored_databases(workspace: Optional[str] = None) -> pd.DataFrame:
+def list_mirrored_databases(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     """
     Shows the mirrored databases within a workspace.
 
@@ -20,8 +21,8 @@ def list_mirrored_databases(workspace: Optional[str] = None) -> pd.DataFrame:
 
     Parameters
     ----------
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -44,7 +45,7 @@ def list_mirrored_databases(workspace: Optional[str] = None) -> pd.DataFrame:
         ]
     )
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
     response = client.get(f"/v1/workspaces/{workspace_id}/mirroredDatabases")
@@ -72,7 +73,7 @@ def list_mirrored_databases(workspace: Optional[str] = None) -> pd.DataFrame:
 
 
 def create_mirrored_database(
-    name: str, description: Optional[str] = None, workspace: Optional[str] = None
+    name: str, description: Optional[str] = None, workspace: Optional[str | UUID] = None
 ):
     """
     Creates a Fabric mirrored database.
@@ -85,13 +86,13 @@ def create_mirrored_database(
         Name of the mirrored database.
     description : str, default=None
         A description of the mirrored database.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     request_body = {"displayName": name}
 
@@ -107,11 +108,13 @@ def create_mirrored_database(
         raise FabricHTTPException(response)
 
     print(
-        f"{icons.green_dot} The '{name}' mirrored database has been created within the '{workspace}' workspace."
+        f"{icons.green_dot} The '{name}' mirrored database has been created within the '{workspace_name}' workspace."
     )
 
 
-def delete_mirrored_database(mirrored_database: str, workspace: Optional[str] = None):
+def delete_mirrored_database(
+    mirrored_database: str, workspace: Optional[str | UUID] = None
+):
     """
     Deletes a mirrored database.
 
@@ -121,16 +124,16 @@ def delete_mirrored_database(mirrored_database: str, workspace: Optional[str] =
     ----------
     mirrored_database: str
         Name of the mirrored database.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     item_id = fabric.resolve_item_id(
-        item_name=mirrored_database, type="MirroredDatabase", workspace=workspace
+        item_name=mirrored_database, type="MirroredDatabase", workspace=workspace_id
     )
 
     client = fabric.FabricRestClient()
@@ -142,12 +145,12 @@ def delete_mirrored_database(mirrored_database: str, workspace: Optional[str] =
         raise FabricHTTPException(response)
 
     print(
-        f"{icons.green_dot} The '{mirrored_database}' mirrored database within the '{workspace}' workspace has been deleted."
+        f"{icons.green_dot} The '{mirrored_database}' mirrored database within the '{workspace_name}' workspace has been deleted."
     )
 
 
 def get_mirroring_status(
-    mirrored_database: str, workspace: Optional[str] = None
+    mirrored_database: str, workspace: Optional[str | UUID] = None
 ) -> str:
     """
     Get the status of the mirrored database.
@@ -158,8 +161,8 @@ def get_mirroring_status(
     ----------
     mirrored_database: str
         Name of the mirrored database.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -169,10 +172,10 @@ def get_mirroring_status(
         The status of a mirrored database.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     item_id = fabric.resolve_item_id(
-        item_name=mirrored_database, type="MirroredDatabase", workspace=workspace
+        item_name=mirrored_database, type="MirroredDatabase", workspace=workspace_id
     )
 
     client = fabric.FabricRestClient()
@@ -187,7 +190,7 @@ def get_mirroring_status(
 
 
 def get_tables_mirroring_status(
-    mirrored_database: str, workspace: Optional[str] = None
+    mirrored_database: str, workspace: Optional[str | UUID] = None
 ) -> pd.DataFrame:
     """
     Gets the mirroring status of the tables.
@@ -198,8 +201,8 @@ def get_tables_mirroring_status(
     ----------
     mirrored_database: str
         Name of the mirrored database.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -209,10 +212,10 @@ def get_tables_mirroring_status(
         A pandas dataframe showing the mirroring status of the tables.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     item_id = fabric.resolve_item_id(
-        item_name=mirrored_database, type="MirroredDatabase", workspace=workspace
+        item_name=mirrored_database, type="MirroredDatabase", workspace=workspace_id
     )
 
     client = fabric.FabricRestClient()
@@ -257,7 +260,7 @@ def get_tables_mirroring_status(
     return df
 
 
-def start_mirroring(mirrored_database: str, workspace: Optional[str] = None):
+def start_mirroring(mirrored_database: str, workspace: Optional[str | UUID] = None):
     """
     Starts the mirroring for a database.
 
@@ -267,16 +270,16 @@ def start_mirroring(mirrored_database: str, workspace: Optional[str] = None):
     ----------
     mirrored_database: str
         Name of the mirrored database.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     item_id = fabric.resolve_item_id(
-        item_name=mirrored_database, type="MirroredDatabase", workspace=workspace
+        item_name=mirrored_database, type="MirroredDatabase", workspace=workspace_id
     )
 
     client = fabric.FabricRestClient()
@@ -288,11 +291,11 @@ def start_mirroring(mirrored_database: str, workspace: Optional[str] = None):
         raise FabricHTTPException(response)
 
     print(
-        f"{icons.green_dot} Mirroring has started for the '{mirrored_database}' database within the '{workspace}' workspace."
+        f"{icons.green_dot} Mirroring has started for the '{mirrored_database}' database within the '{workspace_name}' workspace."
     )
 
 
-def stop_mirroring(mirrored_database: str, workspace: Optional[str] = None):
+def stop_mirroring(mirrored_database: str, workspace: Optional[str | UUID] = None):
     """
     Stops the mirroring for a database.
 
@@ -302,16 +305,16 @@ def stop_mirroring(mirrored_database: str, workspace: Optional[str] = None):
     ----------
     mirrored_database: str
         Name of the mirrored database.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     item_id = fabric.resolve_item_id(
-        item_name=mirrored_database, type="MirroredDatabase", workspace=workspace
+        item_name=mirrored_database, type="MirroredDatabase", workspace=workspace_id
     )
 
     client = fabric.FabricRestClient()
@@ -323,12 +326,12 @@ def stop_mirroring(mirrored_database: str, workspace: Optional[str] = None):
         raise FabricHTTPException(response)
 
     print(
-        f"{icons.green_dot} Mirroring has stopped for the '{mirrored_database}' database within the '{workspace}' workspace."
+        f"{icons.green_dot} Mirroring has stopped for the '{mirrored_database}' database within the '{workspace_name}' workspace."
     )
 
 
 def get_mirrored_database_definition(
-    mirrored_database: str, workspace: Optional[str] = None, decode: bool = True
+    mirrored_database: str, workspace: Optional[str | UUID] = None, decode: bool = True
 ) -> str:
     """
     Obtains the mirrored database definition.
@@ -339,8 +342,8 @@ def get_mirrored_database_definition(
     ----------
     mirrored_database : str
         The name of the mirrored database.
-    workspace : str, default=None
-        The name of the workspace.
+    workspace : str | uuid.UUID, default=None
+        The name or ID 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.
     decode : bool, default=True
@@ -353,9 +356,9 @@ def get_mirrored_database_definition(
         The mirrored database definition.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
     item_id = fabric.resolve_item_id(
-        item_name=mirrored_database, type="MirroredDatabase", workspace=workspace
+        item_name=mirrored_database, type="MirroredDatabase", workspace=workspace_id
     )
     client = fabric.FabricRestClient()
     response = client.post(
@@ -378,7 +381,7 @@ def get_mirrored_database_definition(
 def update_mirrored_database_definition(
     mirrored_database: str,
     mirrored_database_content: dict,
-    workspace: Optional[str] = None,
+    workspace: Optional[str | UUID] = None,
 ):
     """
     Updates an existing notebook with a new definition.
@@ -389,17 +392,17 @@ def update_mirrored_database_definition(
         The name of the mirrored database to be created.
     mirrored_database_content : dict
         The mirrored database definition (not in Base64 format).
-    workspace : str, default=None
-        The name of the workspace.
+    workspace : str | uuid.UUID, default=None
+        The name or ID 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.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
     client = fabric.FabricRestClient()
     payload = base64.b64encode(mirrored_database_content)
     item_id = fabric.resolve_item_id(
-        item_name=mirrored_database, type="MirroredDatabase", workspace=workspace
+        item_name=mirrored_database, type="MirroredDatabase", workspace=workspace_id
     )
 
     request_body = {
@@ -424,5 +427,5 @@ def update_mirrored_database_definition(
     lro(client, response, return_status_code=True)
 
     print(
-        f"{icons.green_dot} The '{mirrored_database}' mirrored database was updated within the '{workspace}' workspace."
+        f"{icons.green_dot} The '{mirrored_database}' mirrored database was updated within the '{workspace_name}' workspace."
     )

sempy_labs/_mirrored_warehouses.py

@@ -6,9 +6,10 @@ from sempy_labs._helper_functions import (
     pagination,
 )
 from sempy.fabric.exceptions import FabricHTTPException
+from uuid import UUID
 
 
-def list_mirrored_warehouses(workspace: Optional[str] = None) -> pd.DataFrame:
+def list_mirrored_warehouses(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     """
     Shows the mirrored warehouses within a workspace.
 
@@ -16,8 +17,8 @@ def list_mirrored_warehouses(workspace: Optional[str] = None) -> pd.DataFrame:
 
     Parameters
     ----------
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -31,7 +32,7 @@ def list_mirrored_warehouses(workspace: Optional[str] = None) -> pd.DataFrame:
         columns=["Mirrored Warehouse Name", "Mirrored Warehouse Id", "Description"]
     )
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
     response = client.get(f"/v1/workspaces/{workspace_id}/mirroredWarehouses")

sempy_labs/_ml_experiments.py

@@ -8,9 +8,10 @@ from sempy_labs._helper_functions import (
     pagination,
 )
 from sempy.fabric.exceptions import FabricHTTPException
+from uuid import UUID
 
 
-def list_ml_experiments(workspace: Optional[str] = None) -> pd.DataFrame:
+def list_ml_experiments(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     """
     Shows the ML experiments within a workspace.
 
@@ -18,8 +19,8 @@ def list_ml_experiments(workspace: Optional[str] = None) -> pd.DataFrame:
 
     Parameters
     ----------
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -31,7 +32,7 @@ def list_ml_experiments(workspace: Optional[str] = None) -> pd.DataFrame:
 
     df = pd.DataFrame(columns=["ML Experiment Name", "ML Experiment Id", "Description"])
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
     response = client.get(f"/v1/workspaces/{workspace_id}/mlExperiments")
@@ -57,7 +58,7 @@ def list_ml_experiments(workspace: Optional[str] = None) -> pd.DataFrame:
 
 
 def create_ml_experiment(
-    name: str, description: Optional[str] = None, workspace: Optional[str] = None
+    name: str, description: Optional[str] = None, workspace: Optional[str | UUID] = None
 ):
     """
     Creates a Fabric ML experiment.
@@ -70,13 +71,13 @@ def create_ml_experiment(
         Name of the ML experiment.
     description : str, default=None
         A description of the environment.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     request_body = {"displayName": name}
 
@@ -91,11 +92,11 @@ def create_ml_experiment(
     lro(client, response, status_codes=[201, 202])
 
     print(
-        f"{icons.green_dot} The '{name}' ML experiment has been created within the '{workspace}' workspace."
+        f"{icons.green_dot} The '{name}' ML experiment has been created within the '{workspace_name}' workspace."
     )
 
 
-def delete_ml_experiment(name: str, workspace: Optional[str] = None):
+def delete_ml_experiment(name: str, workspace: Optional[str | UUID] = None):
     """
     Deletes a Fabric ML experiment.
 
@@ -105,16 +106,16 @@ def delete_ml_experiment(name: str, workspace: Optional[str] = None):
     ----------
     name: str
         Name of the ML experiment.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     item_id = fabric.resolve_item_id(
-        item_name=name, type="MLExperiment", workspace=workspace
+        item_name=name, type="MLExperiment", workspace=workspace_id
     )
 
     client = fabric.FabricRestClient()
@@ -124,5 +125,5 @@ def delete_ml_experiment(name: str, workspace: Optional[str] = None):
         raise FabricHTTPException(response)
 
     print(
-        f"{icons.green_dot} The '{name}' ML experiment within the '{workspace}' workspace has been deleted."
+        f"{icons.green_dot} The '{name}' ML experiment within the '{workspace_name}' workspace has been deleted."
     )

sempy_labs/_ml_models.py

@@ -8,9 +8,10 @@ from sempy_labs._helper_functions import (
     pagination,
 )
 from sempy.fabric.exceptions import FabricHTTPException
+from uuid import UUID
 
 
-def list_ml_models(workspace: Optional[str] = None) -> pd.DataFrame:
+def list_ml_models(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     """
     Shows the ML models within a workspace.
 
@@ -18,8 +19,8 @@ def list_ml_models(workspace: Optional[str] = None) -> pd.DataFrame:
 
     Parameters
     ----------
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -31,7 +32,7 @@ def list_ml_models(workspace: Optional[str] = None) -> pd.DataFrame:
 
     df = pd.DataFrame(columns=["ML Model Name", "ML Model Id", "Description"])
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
     response = client.get(f"/v1/workspaces/{workspace_id}/mlModels")
@@ -57,7 +58,7 @@ def list_ml_models(workspace: Optional[str] = None) -> pd.DataFrame:
 
 
 def create_ml_model(
-    name: str, description: Optional[str] = None, workspace: Optional[str] = None
+    name: str, description: Optional[str] = None, workspace: Optional[str | UUID] = None
 ):
     """
     Creates a Fabric ML model.
@@ -70,13 +71,13 @@ def create_ml_model(
         Name of the ML model.
     description : str, default=None
         A description of the ML model.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     request_body = {"displayName": name}
 
@@ -89,11 +90,11 @@ def create_ml_model(
     lro(client, response, status_codes=[201, 202])
 
     print(
-        f"{icons.green_dot} The '{name}' ML model has been created within the '{workspace}' workspace."
+        f"{icons.green_dot} The '{name}' ML model has been created within the '{workspace_name}' workspace."
     )
 
 
-def delete_ml_model(name: str, workspace: Optional[str] = None):
+def delete_ml_model(name: str, workspace: Optional[str | UUID] = None):
     """
     Deletes a Fabric ML model.
 
@@ -103,13 +104,13 @@ def delete_ml_model(name: str, workspace: Optional[str] = None):
     ----------
     name: str
         Name of the ML model.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     item_id = fabric.resolve_item_id(
         item_name=name, type="MLModel", workspace=workspace
@@ -122,5 +123,5 @@ def delete_ml_model(name: str, workspace: Optional[str] = None):
         raise FabricHTTPException(response)
 
     print(
-        f"{icons.green_dot} The '{name}' ML model within the '{workspace}' workspace has been deleted."
+        f"{icons.green_dot} The '{name}' ML model within the '{workspace_name}' workspace has been deleted."
     )