From 5f9f2aef9b2e7329eae1cda584f3a586b8043994 Mon Sep 17 00:00:00 2001
From: Avik Basu <ab93@users.noreply.github.com>
Date: Mon, 16 Sep 2024 18:40:27 -0700
Subject: [PATCH] fix: lint

Signed-off-by: Avik Basu <ab93@users.noreply.github.com>
---
 numalogic/connectors/wavefront.py | 31 ++++++++++++++++++-------------
 tests/connectors/test_wf.py       |  1 -
 2 files changed, 18 insertions(+), 14 deletions(-)

diff --git a/numalogic/connectors/wavefront.py b/numalogic/connectors/wavefront.py
index c3ee7eb1..703165eb 100644
--- a/numalogic/connectors/wavefront.py
+++ b/numalogic/connectors/wavefront.py
@@ -21,7 +21,8 @@ class WavefrontFetcher(DataFetcher):
         url (str): Wavefront URL.
         api_token (str): Wavefront API token.
 
-    Raises:
+    Raises
+    ------
         ValueError: If API token is not provided.
         WavefrontFetcherError: If there is an error fetching data from Wavefront.
     """
@@ -55,7 +56,8 @@ def _format_results(res: dict) -> pd.DataFrame:
         """Validates and formats the results from the API."""
         if res.get("error_type") is not None:
             raise WavefrontFetcherError(
-                f"Error fetching data from Wavefront: {res.get('error_type')}: {res.get('error_message')}"
+                f"Error fetching data from Wavefront: "
+                f"{res.get('error_type')}: {res.get('error_message')}"
             )
         if res.get("timeseries") is None:
             raise WavefrontFetcherError("No timeseries data found for the query")
@@ -64,8 +66,7 @@ def _format_results(res: dict) -> pd.DataFrame:
             dfs.append(pd.DataFrame(ts["data"], columns=["timestamp", "value"]))
         df = pd.concat(dfs)
         df["timestamp"] = pd.to_datetime(df["timestamp"], unit="s")
-        df = df.set_index("timestamp").sort_index()
-        return df
+        return df.set_index("timestamp").sort_index()
 
     def fetch(
         self,
@@ -79,16 +80,19 @@ def fetch(
         Fetches data from Wavefront as a single metric.
 
         Args:
-            metric (str): Metric to fetch. Example: 'system.cpu.usage'. Do not include the 'ts()' function.
+            metric (str): Metric to fetch. Example: 'system.cpu.usage'.
+                Do not include the 'ts()' function.
             start (datetime): Start time.
             filters (dict): Filters to apply to the query.
             end (datetime): End time. Set to None to fetch data until now.
             granularity (str): Granularity of the data. Default is 'm' (minute).
 
-        Returns:
+        Returns
+        -------
             Dataframe with the fetched data in the format: timestamp (index), value (column).
 
-        Raises:
+        Raises
+        ------
             WavefrontFetcherError: If there is an error fetching data from Wavefront
         """
         start = int(start.timestamp())
@@ -99,7 +103,7 @@ def fetch(
             query = f"ts({metric}, {_filters})"
         else:
             query = f"ts({metric}"
-        LOGGER.info(f"Fetching data from Wavefront for query: {query}")
+        LOGGER.info("Fetching data from Wavefront for query: %s", query)
         res = self._call_api(query, start, end, granularity)
         return self._format_results(res.to_dict())
 
@@ -121,10 +125,12 @@ def raw_fetch(
             end (datetime): End time. Set to None to fetch data until now.
             granularity (str): Granularity of the data. Default is 'm' (minute).
 
-        Returns:
+        Returns
+        -------
             Dataframe with the fetched data in the format: timestamp (index), value (column).
 
-        Raises:
+        Raises
+        ------
             WavefrontFetcherError:
                 - If there is an error fetching data from Wavefront
                 - If there is a key error in the query.
@@ -139,7 +145,6 @@ def raw_fetch(
         ...     end=datetime.now(),
         ... )
         """
-
         start = start.timestamp()
         if end:
             end = end.timestamp()
@@ -147,8 +152,8 @@ def raw_fetch(
         try:
             query = query.format(**filters)
         except KeyError as key_err:
-            raise WavefrontFetcherError(f"Key error in query: {key_err}")
+            raise WavefrontFetcherError(f"Key error in query: {key_err}") from key_err
 
-        LOGGER.info(f"Fetching data from Wavefront for query: {query}")
+        LOGGER.info("Fetching data from Wavefront for query: %s", query)
         qres = self._call_api(query, start, granularity, end)
         return self._format_results(qres.to_dict())
diff --git a/tests/connectors/test_wf.py b/tests/connectors/test_wf.py
index 10648b30..f08b9e16 100644
--- a/tests/connectors/test_wf.py
+++ b/tests/connectors/test_wf.py
@@ -172,7 +172,6 @@ def test_fetch_err_01(wavefront_fetcher, mocker):
 
 
 def test_fetch_err_02(wavefront_fetcher, mocker):
-
     mocker.patch.object(wavefront_fetcher, "_call_api", return_value=DUMMY_OUT_NO_TS)
 
     with pytest.raises(WavefrontFetcherError):