style updates

Author: WillAyd

pandas/core/frame.py

@@ -95,7 +95,7 @@
 )
 from pandas.core.dtypes.missing import isna, notna
 
-from pandas._typing import Axis, Axes, Dtype, FilePathOrBuffer, Level
+from pandas._typing import Axes, Axis, Dtype, FilePathOrBuffer, Level
 from pandas.core import algorithms, common as com, nanops, ops
 from pandas.core.accessor import CachedAccessor
 from pandas.core.arrays import Categorical, ExtensionArray
@@ -2076,7 +2076,7 @@ def to_stata(
             data_label=data_label,
             write_index=write_index,
             variable_labels=variable_labels,
-            **kwargs
+            **kwargs,
         )
         writer.write_file()
 
@@ -2102,7 +2102,7 @@ def to_parquet(
         compression="snappy",
         index=None,
         partition_cols=None,
-        **kwargs
+        **kwargs,
     ):
         """
         Write a DataFrame to the binary parquet format.
@@ -2182,7 +2182,7 @@ def to_parquet(
             compression=compression,
             index=index,
             partition_cols=partition_cols,
-            **kwargs
+            **kwargs,
         )
 
     @Substitution(
@@ -4038,7 +4038,7 @@ def rename(
         mapper: Optional[
             Union[Mapping[Hashable, Hashable], Callable[[Hashable], Hashable]]
         ] = None,
-            *,
+        *,
         index: Optional[
             Union[Mapping[Hashable, Hashable], Callable[[Hashable], Hashable]]
         ] = None,
@@ -4160,8 +4160,16 @@ def rename(
         2  2  5
         4  3  6
         """
-        return super().rename(mapper=mapper, index=index, columns=columns, axis=axis, copy=copy,
-                              inplace=inplace, level=level, errors=errors)
+        return super().rename(
+            mapper=mapper,
+            index=index,
+            columns=columns,
+            axis=axis,
+            copy=copy,
+            inplace=inplace,
+            level=level,
+            errors=errors,
+        )
 
     @Substitution(**_shared_doc_kwargs)
     @Appender(NDFrame.fillna.__doc__)
@@ -4173,7 +4181,7 @@ def fillna(
         inplace=False,
         limit=None,
         downcast=None,
-        **kwargs
+        **kwargs,
     ):
         return super().fillna(
             value=value,
@@ -4182,7 +4190,7 @@ def fillna(
             inplace=inplace,
             limit=limit,
             downcast=downcast,
-            **kwargs
+            **kwargs,
         )
 
     @Appender(_shared_docs["replace"] % _shared_doc_kwargs)
@@ -6629,7 +6637,7 @@ def _gotitem(
         see_also=_agg_summary_and_see_also_doc,
         examples=_agg_examples_doc,
         versionadded="\n.. versionadded:: 0.20.0\n",
-        **_shared_doc_kwargs
+        **_shared_doc_kwargs,
     )
     @Appender(_shared_docs["aggregate"])
     def aggregate(self, func, axis=0, *args, **kwargs):

pandas/core/generic.py

@@ -1019,7 +1019,7 @@ def rename(
         mapper: Optional[
             Union[Mapping[Hashable, Hashable], Callable[[Hashable], Hashable]]
         ] = None,
-            *,
+        *,
         index: Optional[
             Union[Mapping[Hashable, Hashable], Callable[[Hashable], Hashable]]
         ] = None,
@@ -1147,11 +1147,15 @@ def rename(
         if mapper is None and index is None and columns is None:
             raise TypeError("must pass an index to rename")
 
-        if (index is not None or columns is not None):
+        if index is not None or columns is not None:
             if axis is not None:
-                raise TypeError("Cannot specify both 'axis' and any of 'index' or 'columns'")
+                raise TypeError(
+                    "Cannot specify both 'axis' and any of 'index' or 'columns'"
+                )
             elif mapper is not None:
-                raise TypeError("Cannot specify both 'mapper' and any of 'index' or 'columns'")
+                raise TypeError(
+                    "Cannot specify both 'mapper' and any of 'index' or 'columns'"
+                )
         else:
             # use the mapper argument
             if axis in {1, "columns"}:
@@ -1166,7 +1170,7 @@ def rename(
                 continue
 
             axis = self._get_axis(axis_no)
-            baxis = self._get_block_manager_axis(axis_no)            
+            baxis = self._get_block_manager_axis(axis_no)
             f = com.get_rename_function(replacements)
 
             if level is not None:
@@ -1177,7 +1181,9 @@ def rename(
                 indexer = axis.get_indexer_for(replacements)
                 if errors == "raise" and len(indexer[indexer == -1]):
                     missing_labels = [
-                        label for index, label in enumerate(replacements) if indexer[index] == -1
+                        label
+                        for index, label in enumerate(replacements)
+                        if indexer[index] == -1
                     ]
                     raise KeyError("{} not found in axis".format(missing_labels))
 
@@ -7030,7 +7036,7 @@ def interpolate(
         limit_direction="forward",
         limit_area=None,
         downcast=None,
-        **kwargs
+        **kwargs,
     ):
         """
         Interpolate values according to different methods.
@@ -7103,7 +7109,7 @@ def interpolate(
             limit_area=limit_area,
             inplace=inplace,
             downcast=downcast,
-            **kwargs
+            **kwargs,
         )
 
         if inplace:
@@ -7803,7 +7809,7 @@ def groupby(
         group_keys=True,
         squeeze=False,
         observed=False,
-        **kwargs
+        **kwargs,
     ):
         """
         Group DataFrame or Series using a mapper or by a Series of columns.
@@ -7929,7 +7935,7 @@ def groupby(
             group_keys=group_keys,
             squeeze=squeeze,
             observed=observed,
-            **kwargs
+            **kwargs,
         )
 
     def asfreq(self, freq, method=None, how=None, normalize=False, fill_value=None):
@@ -11570,7 +11576,7 @@ def stat_func(
         level=None,
         numeric_only=None,
         min_count=0,
-        **kwargs
+        **kwargs,
     ):
         if name == "sum":
             nv.validate_sum(tuple(), kwargs)

pandas/core/series.py

@@ -53,6 +53,7 @@
 )
 
 import pandas as pd
+from pandas._typing import Axis, Level
 from pandas.core import algorithms, base, generic, nanops, ops
 from pandas.core.accessor import CachedAccessor
 from pandas.core.arrays import ExtensionArray
@@ -78,7 +79,6 @@
 from pandas.core.strings import StringMethods
 from pandas.core.tools.datetimes import to_datetime
 
-from pandas._typing import Axis, Level
 import pandas.io.formats.format as fmt
 import pandas.plotting
 
@@ -4085,12 +4085,12 @@ def rename(
         index: Optional[
             Union[Hashable, Mapping[Hashable, Hashable], Callable[[Hashable], Hashable]]
         ] = None,
-            *,
-            axis: Optional[Axis] = None,
+        *,
+        axis: Optional[Axis] = None,
         copy: bool = True,
         inplace: bool = False,
         level: Optional[Level] = None,
-            errors: str = "ignore"
+        errors: str = "ignore"
     ) -> "Series":
         """
         Alter Series index labels or name.
@@ -4156,7 +4156,9 @@ def rename(
         dtype: int64
         """
         if callable(index) or is_dict_like(index):
-            return super().rename(index, copy=copy, inplace=inplace, level=level, errors=errors)
+            return super().rename(
+                index, copy=copy, inplace=inplace, level=level, errors=errors
+            )
         else:
             return self._set_name(index, inplace=inplace)
 

pandas/tests/frame/test_alter_axes.py

@@ -1379,7 +1379,7 @@ def test_rename_no_mappings_raises(self):
             df.rename(None, columns=None)
 
         with pytest.raises(TypeError, match=msg):
-            df.rename(None, columns=None, index=None)            
+            df.rename(None, columns=None, index=None)
 
     def test_rename_mapper_and_positional_arguments_raises(self):
         # GH 29136