From dd2b6a6c06a24d410eb364eefaf59c08040fdede Mon Sep 17 00:00:00 2001
From: Yashadatt Sawant <yashadatt.sawant@velotio.com>
Date: Fri, 13 Dec 2024 18:30:02 +0530
Subject: [PATCH] Black format files

---
 smsdk/client.py                           | 16 ++++++++++------
 smsdk/smsdk_entities/data_viz/data_viz.py | 10 +++++-----
 smsdk/smsdk_entities/dev_udf/dev_udf.py   | 20 +++++++++-----------
 3 files changed, 24 insertions(+), 22 deletions(-)

diff --git a/smsdk/client.py b/smsdk/client.py
index dab7a37..bb7c57f 100644
--- a/smsdk/client.py
+++ b/smsdk/client.py
@@ -897,13 +897,15 @@ def create_widget_share_link(self, context="/analysis/datavis", **kwargs):
         base_url = get_url(
             self.config["protocol"], self.tenant, self.config["site.domain"]
         )
-        return dataViz(self.session, base_url).create_widget_share_link(context,**kwargs)
+        return dataViz(self.session, base_url).create_widget_share_link(
+            context, **kwargs
+        )
 
     @version_check_decorator
     def fetch_list_of_udf(self):
         base_url = get_url(
-                self.config["protocol"], self.tenant, self.config["site.domain"]
-            )
+            self.config["protocol"], self.tenant, self.config["site.domain"]
+        )
         # load the entity class and initialize it
         cls = smsdkentities.get("dev_udf")(self.session, base_url)
         udf_list = getattr(cls, "get_list_of_udf")()
@@ -919,9 +921,11 @@ def get_udf_data(self, udf_name, **params):
                 )
                 # load the entity class and initialize it
                 cls = smsdkentities.get("dev_udf")(self.session, base_url)
-                udf_data = getattr(cls, "get_udf_data")(udf_name,params)
+                udf_data = getattr(cls, "get_udf_data")(udf_name, params)
                 return udf_data
             else:
-                log.error(f'UDF named "{udf_name}" does not exist. Please check the name again')
+                log.error(
+                    f'UDF named "{udf_name}" does not exist. Please check the name again'
+                )
         else:
-            log.error('Name of user defined function is required')
+            log.error("Name of user defined function is required")
diff --git a/smsdk/smsdk_entities/data_viz/data_viz.py b/smsdk/smsdk_entities/data_viz/data_viz.py
index 18d2c07..3455b40 100644
--- a/smsdk/smsdk_entities/data_viz/data_viz.py
+++ b/smsdk/smsdk_entities/data_viz/data_viz.py
@@ -81,7 +81,7 @@ def create_share_link(
                 "selectedTimeZone": time_selection["time_zone"],
             }
         else:
-            dateRange={}
+            dateRange = {}
         url_params["state"] = {
             "dataModel": model,
             "asset": asset,
@@ -157,7 +157,7 @@ def create_share_link(
                 url_params["state"]["yAxis"] = yAxis
         response = getattr(self.session, "post")(url, json=url_params)
         return "{}/#{}/s/{}".format(
-            self.base_url,url_params["context"], response.json()["state_hash"]
+            self.base_url, url_params["context"], response.json()["state_hash"]
         )
 
     @mod_util
@@ -185,7 +185,7 @@ def get_dashboard_widget_data(self, model, *args, **kwargs):
         return records
 
     @mod_util
-    def create_widget_share_link(self,context,**kwargs):
+    def create_widget_share_link(self, context, **kwargs):
         url = "{}{}".format(self.base_url, ENDPOINTS["DataViz"]["share_link"])
         url_params = {}
         url_params["state_hash"] = str(uuid.uuid4())[:8]
@@ -194,8 +194,8 @@ def create_widget_share_link(self,context,**kwargs):
             url_params["in_use_workspace"] = self.session.headers["X-Sm-Workspace-Id"]
         except:
             pass
-        url_params["state"]=kwargs
+        url_params["state"] = kwargs
         response = getattr(self.session, "post")(url, json=url_params)
         return "{}/#{}/s/{}".format(
             self.base_url, url_params["context"], response.json()["state_hash"]
-        )
\ No newline at end of file
+        )
diff --git a/smsdk/smsdk_entities/dev_udf/dev_udf.py b/smsdk/smsdk_entities/dev_udf/dev_udf.py
index fd2a61a..704d71f 100644
--- a/smsdk/smsdk_entities/dev_udf/dev_udf.py
+++ b/smsdk/smsdk_entities/dev_udf/dev_udf.py
@@ -42,10 +42,10 @@ def get_list_of_udf(self):
         """
         url = "{}{}".format(self.base_url, ENDPOINTS["UDF_dev"]["list_url"])
         html_content = self.session.get(url).text
-        soup = BeautifulSoup(html_content, 'html.parser')
-        table = soup.find('table')
-        rows = table.find('tbody').find_all('tr')
-        list_of_udfs = [row.find_all('td')[0].text for row in rows]
+        soup = BeautifulSoup(html_content, "html.parser")
+        table = soup.find("table")
+        rows = table.find("tbody").find_all("tr")
+        list_of_udfs = [row.find_all("td")[0].text for row in rows]
         return list_of_udfs
 
     @mod_util
@@ -54,15 +54,13 @@ def get_udf_data(self, udf_name, params):
         Utility function to get the data after executing udf notebook
         """
         url = "{}{}".format(self.base_url, ENDPOINTS["UDF_dev"]["url"])
-        payload = {
-            "name": udf_name
-            }
+        payload = {"name": udf_name}
         if params:
-            payload['parameters']= params
+            payload["parameters"] = params
         results = self.session.post(url, json=payload)
         time.sleep(10)
-        async_task_id = results.json().get('response').get('task_id')
-        results = self.session.get(url+'/'+async_task_id).json()
+        async_task_id = results.json().get("response").get("task_id")
+        results = self.session.get(url + "/" + async_task_id).json()
         time.sleep(10)
-        data = results.get('response').get('meta')[0].get('data')
+        data = results.get("response").get("meta")[0].get("data")
         return data