fix: lint

ni · Nov 26, 2024 · 2706f58 · 2706f58
1 parent 204ea7d
commit 2706f58
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/nisystemlink/clients/dataframe/utilities/_pandas_dataframe_operations.py b/nisystemlink/clients/dataframe/utilities/_pandas_dataframe_operations.py
@@ -91,7 +91,9 @@ def create_table_with_data_from_pandas_df(
     for start_row in range(0, num_rows, batch_size):
         end_row = min(start_row + batch_size, num_rows)
         batch_df = df.iloc[start_row:end_row]
-        append_pandas_df_to_table(client, table_id, batch_df, end_of_data=(end_row == num_rows))
+        append_pandas_df_to_table(
+            client, table_id, batch_df, end_of_data=(end_row == num_rows)
+        )
 
     return table_id
 

diff --git a/tests/integration/dataframe/test_pandas_utility.py b/tests/integration/dataframe/test_pandas_utility.py
@@ -203,12 +203,14 @@ def test__query_decimated_table_data_as_pandas_df__succeeds(
             ),
             index=True,
         )
-        data = [
-            ["1", "2", "3"],
-            ["7", "8", "9"],
-        ]
-        columns = ["a", "b", "c"]
-        expected_df = pd.DataFrame(data=data, columns=columns)
+
+        expected_df = pd.DataFrame(
+            data=[
+                ["1", "2", "3"],
+                ["7", "8", "9"],
+            ],
+            columns=["a", "b", "c"],
+        )
         expected_df.set_index("a", inplace=True)
 
         assert (response.values == expected_df.values).all()