example with line length limit of 79

Author: jorisvandenbossche

pandas/__init__.py

@@ -14,7 +14,8 @@
 
 if missing_dependencies:
     raise ImportError(
-        "Unable to import required dependencies:\n" + "\n".join(missing_dependencies)
+        "Unable to import required dependencies:\n"
+        + "\n".join(missing_dependencies)
     )
 del hard_dependencies, dependency, missing_dependencies
 
@@ -27,7 +28,11 @@
 )
 
 try:
-    from pandas._libs import hashtable as _hashtable, lib as _lib, tslib as _tslib
+    from pandas._libs import (
+        hashtable as _hashtable,
+        lib as _lib,
+        tslib as _tslib,
+    )
 except ImportError as e:  # pragma: no cover
     # hack but overkill to use re
     module = str(e).replace("cannot import name ", "")

pandas/_config/config.py

@@ -55,7 +55,9 @@
 import warnings
 
 DeprecatedOption = namedtuple("DeprecatedOption", "key msg rkey removal_ver")
-RegisteredOption = namedtuple("RegisteredOption", "key defval doc validator cb")
+RegisteredOption = namedtuple(
+    "RegisteredOption", "key defval doc validator cb"
+)
 
 # holds deprecated option metdata
 _deprecated_options = {}  # type: Dict[str, DeprecatedOption]
@@ -110,7 +112,9 @@ def _set_option(*args, **kwargs):
     # must at least 1 arg deal with constraints later
     nargs = len(args)
     if not nargs or nargs % 2 != 0:
-        raise ValueError("Must provide an even number of non-keyword " "arguments")
+        raise ValueError(
+            "Must provide an even number of non-keyword " "arguments"
+        )
 
     # default to false
     silent = kwargs.pop("silent", False)
@@ -236,7 +240,9 @@ def __call__(self, *args, **kwds):
     def __doc__(self):
         opts_desc = _describe_option("all", _print_desc=False)
         opts_list = pp_options_list(list(_registered_options.keys()))
-        return self.__doc_tmpl__.format(opts_desc=opts_desc, opts_list=opts_list)
+        return self.__doc_tmpl__.format(
+            opts_desc=opts_desc, opts_list=opts_list
+        )
 
 
 _get_option_tmpl = """
@@ -395,13 +401,16 @@ class option_context:
     def __init__(self, *args):
         if not (len(args) % 2 == 0 and len(args) >= 2):
             raise ValueError(
-                "Need to invoke as" " option_context(pat, val, [(pat, val), ...])."
+                "Need to invoke as"
+                " option_context(pat, val, [(pat, val), ...])."
             )
 
         self.ops = list(zip(args[::2], args[1::2]))
 
     def __enter__(self):
-        self.undo = [(pat, _get_option(pat, silent=True)) for pat, val in self.ops]
+        self.undo = [
+            (pat, _get_option(pat, silent=True)) for pat, val in self.ops
+        ]
 
         for pat, val in self.ops:
             _set_option(pat, val, silent=True)
@@ -623,7 +632,9 @@ def _warn_if_deprecated(key):
         else:
             msg = "'{key}' is deprecated".format(key=key)
             if d.removal_ver:
-                msg += " and will be removed in {version}".format(version=d.removal_ver)
+                msg += " and will be removed in {version}".format(
+                    version=d.removal_ver
+                )
             if d.rkey:
                 msg += ", please use '{rkey}' instead.".format(rkey=d.rkey)
             else:

pandas/_config/display.py

@@ -52,5 +52,8 @@ def detect_console_encoding():
 
 with cf.config_prefix("display"):
     cf.register_option(
-        "encoding", detect_console_encoding(), pc_encoding_doc, validator=cf.is_text
+        "encoding",
+        detect_console_encoding(),
+        pc_encoding_doc,
+        validator=cf.is_text,
     )

pandas/_config/localization.py

@@ -108,7 +108,9 @@ def _default_locale_getter():
     return raw_locales
 
 
-def get_locales(prefix=None, normalize=True, locale_getter=_default_locale_getter):
+def get_locales(
+    prefix=None, normalize=True, locale_getter=_default_locale_getter
+):
     """
     Get all the locales that are available on the system.
 

pandas/_typing.py

@@ -25,7 +25,9 @@
     np.ndarray,
 )
 ArrayLike = TypeVar("ArrayLike", ABCExtensionArray, np.ndarray)
-DatetimeLikeScalar = TypeVar("DatetimeLikeScalar", Period, Timestamp, Timedelta)
+DatetimeLikeScalar = TypeVar(
+    "DatetimeLikeScalar", Period, Timestamp, Timedelta
+)
 Dtype = Union[str, np.dtype, ExtensionDtype]
 FilePathOrBuffer = Union[str, Path, IO[AnyStr]]
 

pandas/_version.py

@@ -103,7 +103,9 @@ def versions_from_parentdir(parentdir_prefix, root, verbose):
             print(
                 "guessing rootdir is '{root}', but '{dirname}' "
                 "doesn't start with prefix '{parentdir_prefix}'".format(
-                    root=root, dirname=dirname, parentdir_prefix=parentdir_prefix
+                    root=root,
+                    dirname=dirname,
+                    parentdir_prefix=parentdir_prefix,
                 )
             )
         raise NotThisMethod("rootdir doesn't start with parentdir_prefix")
@@ -249,7 +251,9 @@ def git_pieces_from_vcs(tag_prefix, root, verbose, run_command=run_command):
         # tag
         full_tag = mo.group(1)
         if not full_tag.startswith(tag_prefix):
-            fmt = "tag '{full_tag}' doesn't start with prefix " "'{tag_prefix}'"
+            fmt = (
+                "tag '{full_tag}' doesn't start with prefix " "'{tag_prefix}'"
+            )
             msg = fmt.format(full_tag=full_tag, tag_prefix=tag_prefix)
             if verbose:
                 print(msg)
@@ -267,7 +271,9 @@ def git_pieces_from_vcs(tag_prefix, root, verbose, run_command=run_command):
     else:
         # HEX: no tags
         pieces["closest-tag"] = None
-        count_out = run_command(GITS, ["rev-list", "HEAD", "--count"], cwd=root)
+        count_out = run_command(
+            GITS, ["rev-list", "HEAD", "--count"], cwd=root
+        )
         pieces["distance"] = int(count_out)  # total number of commits
 
     return pieces
@@ -296,7 +302,9 @@ def render_pep440(pieces):
                 rendered += ".dirty"
     else:
         # exception #1
-        rendered = "0+untagged.{:d}.g{}".format(pieces["distance"], pieces["short"])
+        rendered = "0+untagged.{:d}.g{}".format(
+            pieces["distance"], pieces["short"]
+        )
         if pieces["dirty"]:
             rendered += ".dirty"
     return rendered
@@ -446,7 +454,9 @@ def get_versions():
     verbose = cfg.verbose
 
     try:
-        return git_versions_from_keywords(get_keywords(), cfg.tag_prefix, verbose)
+        return git_versions_from_keywords(
+            get_keywords(), cfg.tag_prefix, verbose
+        )
     except NotThisMethod:
         pass
 

pandas/api/extensions/__init__.py

@@ -6,4 +6,7 @@
 )
 from pandas.core.algorithms import take  # noqa
 from pandas.core.arrays import ExtensionArray, ExtensionScalarOpsMixin  # noqa
-from pandas.core.dtypes.dtypes import ExtensionDtype, register_extension_dtype  # noqa
+from pandas.core.dtypes.dtypes import (
+    ExtensionDtype,
+    register_extension_dtype,
+)  # noqa

pandas/compat/_optional.py

@@ -44,12 +44,17 @@ def _get_version(module: types.ModuleType) -> str:
         version = getattr(module, "__VERSION__", None)
 
     if version is None:
-        raise ImportError("Can't determine version for {}".format(module.__name__))
+        raise ImportError(
+            "Can't determine version for {}".format(module.__name__)
+        )
     return version
 
 
 def import_optional_dependency(
-    name: str, extra: str = "", raise_on_missing: bool = True, on_version: str = "raise"
+    name: str,
+    extra: str = "",
+    raise_on_missing: bool = True,
+    on_version: str = "raise",
 ):
     """
     Import an optional dependency.
@@ -99,7 +104,9 @@ def import_optional_dependency(
         if distutils.version.LooseVersion(version) < minimum_version:
             assert on_version in {"warn", "raise", "ignore"}
             msg = version_message.format(
-                minimum_version=minimum_version, name=name, actual_version=version
+                minimum_version=minimum_version,
+                name=name,
+                actual_version=version,
             )
             if on_version == "warn":
                 warnings.warn(msg, UserWarning)

pandas/compat/numpy/function.py

@@ -33,13 +33,17 @@
 
 
 class CompatValidator:
-    def __init__(self, defaults, fname=None, method=None, max_fname_arg_count=None):
+    def __init__(
+        self, defaults, fname=None, method=None, max_fname_arg_count=None
+    ):
         self.fname = fname
         self.method = method
         self.defaults = defaults
         self.max_fname_arg_count = max_fname_arg_count
 
-    def __call__(self, args, kwargs, fname=None, max_fname_arg_count=None, method=None):
+    def __call__(
+        self, args, kwargs, fname=None, max_fname_arg_count=None, method=None
+    ):
         if args or kwargs:
             fname = self.fname if fname is None else fname
             max_fname_arg_count = (
@@ -59,7 +63,8 @@ def __call__(self, args, kwargs, fname=None, max_fname_arg_count=None, method=No
                 )
             else:
                 raise ValueError(
-                    "invalid validation method " "'{method}'".format(method=method)
+                    "invalid validation method "
+                    "'{method}'".format(method=method)
                 )
 
 
@@ -108,7 +113,9 @@ def validate_argmax_with_skipna(skipna, args, kwargs):
     return skipna
 
 
-ARGSORT_DEFAULTS = OrderedDict()  # type: OrderedDict[str, Optional[Union[int, str]]]
+ARGSORT_DEFAULTS = (
+    OrderedDict()
+)  # type: OrderedDict[str, Optional[Union[int, str]]]
 ARGSORT_DEFAULTS["axis"] = -1
 ARGSORT_DEFAULTS["kind"] = "quicksort"
 ARGSORT_DEFAULTS["order"] = None
@@ -128,7 +135,10 @@ def validate_argmax_with_skipna(skipna, args, kwargs):
 ARGSORT_DEFAULTS_KIND["axis"] = -1
 ARGSORT_DEFAULTS_KIND["order"] = None
 validate_argsort_kind = CompatValidator(
-    ARGSORT_DEFAULTS_KIND, fname="argsort", max_fname_arg_count=0, method="both"
+    ARGSORT_DEFAULTS_KIND,
+    fname="argsort",
+    max_fname_arg_count=0,
+    method="both",
 )
 
 
@@ -241,7 +251,9 @@ def validate_cum_func_with_skipna(skipna, args, kwargs, name):
     ROUND_DEFAULTS, fname="round", method="both", max_fname_arg_count=1
 )
 
-SORT_DEFAULTS = OrderedDict()  # type: OrderedDict[str, Optional[Union[int, str]]]
+SORT_DEFAULTS = (
+    OrderedDict()
+)  # type: OrderedDict[str, Optional[Union[int, str]]]
 SORT_DEFAULTS["axis"] = -1
 SORT_DEFAULTS["kind"] = "quicksort"
 SORT_DEFAULTS["order"] = None
@@ -275,11 +287,15 @@ def validate_cum_func_with_skipna(skipna, args, kwargs, name):
     MEDIAN_DEFAULTS, fname="median", method="both", max_fname_arg_count=1
 )
 
-STAT_DDOF_FUNC_DEFAULTS = OrderedDict()  # type: OrderedDict[str, Optional[bool]]
+STAT_DDOF_FUNC_DEFAULTS = (
+    OrderedDict()
+)  # type: OrderedDict[str, Optional[bool]]
 STAT_DDOF_FUNC_DEFAULTS["dtype"] = None
 STAT_DDOF_FUNC_DEFAULTS["out"] = None
 STAT_DDOF_FUNC_DEFAULTS["keepdims"] = False
-validate_stat_ddof_func = CompatValidator(STAT_DDOF_FUNC_DEFAULTS, method="kwargs")
+validate_stat_ddof_func = CompatValidator(
+    STAT_DDOF_FUNC_DEFAULTS, method="kwargs"
+)
 
 TAKE_DEFAULTS = OrderedDict()  # type: OrderedDict[str, Optional[str]]
 TAKE_DEFAULTS["out"] = None

pandas/compat/pickle_compat.py

@@ -57,7 +57,10 @@ def load_reduce(self):
 
 # If classes are moved, provide compat here.
 _class_locations_map = {
-    ("pandas.core.sparse.array", "SparseArray"): ("pandas.core.arrays", "SparseArray"),
+    ("pandas.core.sparse.array", "SparseArray"): (
+        "pandas.core.arrays",
+        "SparseArray",
+    ),
     # 15477
     #
     # TODO: When FrozenNDArray is removed, add
@@ -74,7 +77,10 @@ def load_reduce(self):
         "pandas.core.indexes.frozen",
         "FrozenNDArray",
     ),
-    ("pandas.core.base", "FrozenList"): ("pandas.core.indexes.frozen", "FrozenList"),
+    ("pandas.core.base", "FrozenList"): (
+        "pandas.core.indexes.frozen",
+        "FrozenList",
+    ),
     # 10890
     ("pandas.core.series", "TimeSeries"): ("pandas.core.series", "Series"),
     ("pandas.sparse.series", "SparseTimeSeries"): (
@@ -86,7 +92,10 @@ def load_reduce(self):
     ("pandas.tslib", "Timestamp"): ("pandas._libs.tslib", "Timestamp"),
     # 18543 moving period
     ("pandas._period", "Period"): ("pandas._libs.tslibs.period", "Period"),
-    ("pandas._libs.period", "Period"): ("pandas._libs.tslibs.period", "Period"),
+    ("pandas._libs.period", "Period"): (
+        "pandas._libs.tslibs.period",
+        "Period",
+    ),
     # 18014 moved __nat_unpickle from _libs.tslib-->_libs.tslibs.nattype
     ("pandas.tslib", "__nat_unpickle"): (
         "pandas._libs.tslibs.nattype",
@@ -109,14 +118,23 @@ def load_reduce(self):
         "pandas.core.sparse.frame",
         "SparseDataFrame",
     ),
-    ("pandas.indexes.base", "_new_Index"): ("pandas.core.indexes.base", "_new_Index"),
+    ("pandas.indexes.base", "_new_Index"): (
+        "pandas.core.indexes.base",
+        "_new_Index",
+    ),
     ("pandas.indexes.base", "Index"): ("pandas.core.indexes.base", "Index"),
     ("pandas.indexes.numeric", "Int64Index"): (
         "pandas.core.indexes.numeric",
         "Int64Index",
     ),
-    ("pandas.indexes.range", "RangeIndex"): ("pandas.core.indexes.range", "RangeIndex"),
-    ("pandas.indexes.multi", "MultiIndex"): ("pandas.core.indexes.multi", "MultiIndex"),
+    ("pandas.indexes.range", "RangeIndex"): (
+        "pandas.core.indexes.range",
+        "RangeIndex",
+    ),
+    ("pandas.indexes.multi", "MultiIndex"): (
+        "pandas.core.indexes.multi",
+        "MultiIndex",
+    ),
     ("pandas.tseries.index", "_new_DatetimeIndex"): (
         "pandas.core.indexes.datetimes",
         "_new_DatetimeIndex",
@@ -130,7 +148,10 @@ def load_reduce(self):
         "PeriodIndex",
     ),
     # 19269, arrays moving
-    ("pandas.core.categorical", "Categorical"): ("pandas.core.arrays", "Categorical"),
+    ("pandas.core.categorical", "Categorical"): (
+        "pandas.core.arrays",
+        "Categorical",
+    ),
     # 19939, add timedeltaindex, float64index compat from 15998 move
     ("pandas.tseries.tdi", "TimedeltaIndex"): (
         "pandas.core.indexes.timedeltas",