code formatting tweaks

CountESS-Project · Oct 18, 2023 · 8056b2b · 8056b2b
1 parent ebbde41
commit 8056b2b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/countess/plugins/csv.py b/countess/plugins/csv.py
@@ -182,9 +182,7 @@ def process(self, data: pd.DataFrame, source: str, logger: Logger):
         # I've set "index=True" below to emit the indexes
 
         drop_index = data.index.name is None and data.index.names[0] is None
-        dataframe = flatten_columns(
-            data.reset_index(drop=drop_index)
-        )
+        dataframe = flatten_columns(data.reset_index(drop=drop_index))
 
         # if this is our first dataframe to write then decide whether to
         # include the header or not.

diff --git a/countess/utils/pandas.py b/countess/utils/pandas.py
@@ -68,8 +68,7 @@ def flatten_columns(dataframe: pd.DataFrame, inplace: bool = False) -> pd.DataFr
         dataframe = dataframe.copy()
 
     dataframe.columns = [
-        "__".join(a) if type(a) is tuple else a
-        for a in dataframe.columns.to_flat_index()
-    ] # type: ignore [assignment]
+        "__".join(a) if type(a) is tuple else a for a in dataframe.columns.to_flat_index()
+    ]  # type: ignore [assignment]
 
     return dataframe