black

Author: m-kovalsky

src/sempy_labs/report/_endorsement.py

@@ -33,7 +33,9 @@ def set_endorsement(
     """
 
     (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
-    (report_name, report_id) = resolve_item_name_and_id(item=report, type='Report', workspace=workspace_id)
+    (report_name, report_id) = resolve_item_name_and_id(
+        item=report, type="Report", workspace=workspace_id
+    )
 
     import notebookutils
 
@@ -50,7 +52,9 @@ def set_endorsement(
     }
 
     if endorsement not in endorsement_mapping:
-        raise ValueError("Endorsement must be either 'None', 'Promoted', 'Certified', or 'Master data'.")
+        raise ValueError(
+            "Endorsement must be either 'None', 'Promoted', 'Certified', or 'Master data'."
+        )
     stage = endorsement_mapping.get(endorsement)
     payload = {"stage": stage}
 

src/sempy_labs/semantic_model/_copilot.py

@@ -96,7 +96,9 @@ def set_endorsement(
     }
 
     if endorsement not in endorsement_mapping:
-        raise ValueError("Endorsement must be either 'None', 'Promoted', 'Certified', or 'Master data'.")
+        raise ValueError(
+            "Endorsement must be either 'None', 'Promoted', 'Certified', or 'Master data'."
+        )
 
     stage = endorsement_mapping.get(endorsement)
     payload = {"stage": stage}
@@ -116,7 +118,9 @@ def set_endorsement(
 
 
 @log
-def make_discoverable(dataset: str | UUID, make_discoverable: bool, workspace: Optional[str | UUID] = None):
+def make_discoverable(
+    dataset: str | UUID, make_discoverable: bool, workspace: Optional[str | UUID] = None
+):
     """
     Allow users without access to this semantic model to discover it and request permissions to access the data. The semantic model must be endorsed as 'Promoted', 'Certified', or 'Master data' to be made discoverable.
 
@@ -161,4 +165,4 @@ def make_discoverable(dataset: str | UUID, make_discoverable: bool, workspace: O
     else:
         print(
             f"{icons.green_dot} The '{item_name}' semantic model within the '{workspace_name}' workspace is no longer discoverable in the gallery."
-        )
+        )

src/sempy_labs/tom/_model.py

@@ -132,8 +132,7 @@ def __init__(self, dataset, workspace, readonly):
 
             self._tom_server = TOM.Server()
             get_access_token = partial(
-                refresh_tom_access_token,
-                credential=ConstantTokenCredential(token)
+                refresh_tom_access_token, credential=ConstantTokenCredential(token)
             )
             self._tom_server.AccessToken = get_access_token(None)
             self._tom_server.OnAccessTokenExpired = Func[AccessToken, AccessToken](

src/sempy_labs/variable_library/_functions.py

@@ -670,9 +670,9 @@ def update_variable(
     variable_library: str | UUID,
     name: str,
     new_name: Optional[str] = None,
-    new_type: Literal["Boolean", "DateTime", "Number", "Integer", "String"] = None,
-    new_value: Optional[str] = None,
-    new_note: Optional[str] = None,
+    type: Literal["Boolean", "DateTime", "Number", "Integer", "String"] = None,
+    value: Optional[str] = None,
+    note: Optional[str] = None,
     workspace: Optional[str | UUID] = None,
 ):
     """
@@ -687,13 +687,13 @@ def update_variable(
     name : str
         Name of the variable.
     new_name : str, default=None
-        New name of the variable.
-    new_type : Literal["Boolean", "DateTime", "Number", "Integer", "String"], default=None
-        New type of the variable. Valid types are: "Boolean", "DateTime", "Number", "Integer", "String".
-    new_value : str, default=None
-        New value of the variable.
-    new_note : str, default=None
-        New note of the variable.
+        New name of the variable. Defaults to None which keeps the same name.
+    type : 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.
+    note : str, default=None
+        New note of the variable. Defaults to None which keeps the same note.
     workspace : str | uuid.UUID, default=None
         The Fabric workspace name or ID.
         Defaults to None which resolves to the workspace of the attached lakehouse
@@ -712,7 +712,7 @@ def update_variable(
         return_dataframe=False,
     )
 
-    if new_name is None and new_type is None and new_value is None and new_note is None:
+    if new_name is None and type is None and value is None and note is None:
         print(
             f"{icons.info} No updates provided for the variable '{name}' in the '{item_name}' variable library within the '{workspace_name}' workspace."
         )
@@ -733,12 +733,12 @@ def update_variable(
                 if variable.get("name") == name:
                     if new_name is not None:
                         variable["name"] = new_name
-                    if new_type is not None:
-                        variable["type"] = new_type
-                    if new_note is not None:
-                        variable["note"] = new_note
-                    if new_value is not None:
-                        variable["value"] = new_value
+                    if type is not None:
+                        variable["type"] = type
+                    if note is not None:
+                        variable["note"] = note
+                    if value is not None:
+                        variable["value"] = value
                     break
             part["payload"] = _encode_b64(payload)