pyannote.database 5.0.0__tar.gz → 5.1.0__tar.gz

{pyannote.database-5.0.0/pyannote.database.egg-info → pyannote.database-5.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyannote.database
-Version: 5.0.0
+Version: 5.1.0
 Summary: Interface to multimedia databases and experimental protocols
 Home-page: http://pyannote.github.io/
 Author: Hervé Bredin
@@ -9,10 +9,9 @@ Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Science/Research
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Natural Language :: English
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.6
-Classifier: Programming Language :: Python :: 3.7
-Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
 Classifier: Topic :: Scientific/Engineering
 Provides-Extra: testing
 Provides-Extra: doc

{pyannote.database-5.0.0 → pyannote.database-5.1.0}/pyannote/database/_version.py

@@ -11,8 +11,8 @@ version_json = '''
 {
  "dirty": false,
  "error": null,
- "full-revisionid": "d8f589d27d41cad27d25510e3d6c051ed79b68c7",
- "version": "5.0.0"
+ "full-revisionid": "b1fc3492e7ea8aa822f1c0993fcd38c485a6b61e",
+ "version": "5.1.0"
 }
 '''  # END VERSION_JSON
 

{pyannote.database-5.0.0 → pyannote.database-5.1.0}/pyannote/database/custom.py

@@ -51,6 +51,7 @@ from . import protocol as protocol_module
 from pyannote.database.protocol.protocol import ProtocolFile
 import yaml
 import warnings
+from numbers import Number
 from typing import Text, Dict, Callable, Any, Union
 import functools
 
@@ -115,6 +116,12 @@ def Template(template: Text, database_yml: Path) -> Callable[[ProtocolFile], Any
     return load
 
 
+def NumericValue(value):
+    def load(current_file: ProtocolFile):
+        return value
+    return load
+
+
 def resolve_path(path: Path, database_yml: Path) -> Path:
     """Resolve path
 
@@ -211,6 +218,10 @@ def gather_loaders(
         if key == "uri" or key == "trial":
             continue
 
+        if isinstance(value, Number):
+            lazy_loader[key] = NumericValue(value)
+            continue
+
         # check whether value (path) contains placeholders such as {uri} or {subset}
         _, placeholders, _, _ = zip(*string.Formatter().parse(value))
         is_template = len(set(placeholders) - set([None])) > 0
@@ -442,16 +453,18 @@ def create_protocol(
             metadata["classes"] = protocol_entries.pop("classes")
 
     if issubclass(base_class, SpeakerDiarizationProtocol):
-        if "scope" in protocol_entries:
-            scope = protocol_entries.pop("scope")
-        else:
-            scope = "file"
+        scope = protocol_entries.pop("scope", None)
+
+        if scope is None and database != "X":
             msg = (
                 f"'{database}.{task}.{protocol}' found in {database_yml} does not define "
                 f"the 'scope' of speaker labels (file, database, or global). Setting it to 'file'."
             )   
             print(msg)
-        metadata["scope"] = scope
+            metadata["scope"] = "file"
+
+        else:
+            metadata["scope"] = scope
 
     methods = dict()
     for subset, subset_entries in protocol_entries.items():

{pyannote.database-5.0.0 → pyannote.database-5.1.0}/pyannote/database/registry.py

@@ -39,11 +39,13 @@ from .custom import create_protocol, get_init
 from .database import Database
 import yaml
 
+
 # controls what to do in case of protocol name conflict
 class LoadingMode(Enum):
     OVERRIDE = 0  # override existing protocol
-    KEEP = 1      # keep existing protocol
-    ERROR = 2     # raise an error 
+    KEEP = 1  # keep existing protocol
+    ERROR = 2  # raise an error
+
 
 # To ease the understanding of future me, all comments inside Registry codebase
 # assume the existence of the following database.yml files.
@@ -52,7 +54,7 @@ class LoadingMode(Enum):
 # Content of /path/to/first/database.yml
 # ======================================
 # Databases:
-#     DatabaseA: 
+#     DatabaseA:
 #         - relative/path/A/trn/{uri}.wav
 #         - relative/path/A/dev/{uri}.wav
 #         - relative/path/A/tst/{uri}.wav
@@ -95,7 +97,7 @@ class LoadingMode(Enum):
 #     DatabaseC:
 #         SpeakerDiarization:
 #             Protocol:
-#                 ...    
+#                 ...
 
 
 class Registry:
@@ -109,13 +111,12 @@ class Registry:
     """
 
     def __init__(self) -> None:
-
         # Mapping of database.yml paths to their config in a dictionary
         # Example after loading both database.yml:
         #   {"/path/to/first/database.yml": {
         #           "Databases":{
         #               "DatabaseA": ["relative/path/A/trn/{uri}.wav", "relative/path/A/dev/{uri}.wav", relative/path/A/tst/{uri}.wav]
-        #               "DatabaseB": "/absolute/path/B/{uri}.wav" 
+        #               "DatabaseB": "/absolute/path/B/{uri}.wav"
         #           },
         #           "Protocols":{
         #               "DatabaseA":{
@@ -134,7 +135,7 @@ class Registry:
         #    "/path/to/second/database.yml": {
         #           "Databases":{
         #               "DatabaseC": /absolute/path/C/{uri}.wav
-        #               "DatabaseB": "/absolute/path/B/{uri}.wav" 
+        #               "DatabaseB": "/absolute/path/B/{uri}.wav"
         #           },
         #           "Protocols":{
         #               "DatabaseB":{"SpeakerDiarization": {"Protocol": {...}}},
@@ -144,7 +145,6 @@ class Registry:
         #   }
         self.configs: Dict[Path, Dict] = dict()
 
-
         # Content of the "Database" root item (= where to find file content)
         # Example after loading both database.yml:
         #   {
@@ -176,7 +176,7 @@ class Registry:
         Parameters
         ----------
         path : str or Path
-            Path to YAML configuration file. 
+            Path to YAML configuration file.
         mode : LoadingMode, optional
             Controls how to handle conflicts in protocol names.
             Defaults to overriding the existing protocol.
@@ -210,7 +210,7 @@ class Registry:
         # make path absolute
         database_yml = Path(database_yml).expanduser().resolve()
 
-        # stop here if configuration file is already being loaded 
+        # stop here if configuration file is already being loaded
         # (possibly because of circular requirements)
         if database_yml in loading:
             return
@@ -221,9 +221,9 @@ class Registry:
         # load configuration
         with open(database_yml, "r") as f:
             config = yaml.load(f, Loader=yaml.SafeLoader)
-        
+
         # load every requirement
-        requirements = config.pop('Requirements', list())
+        requirements = config.pop("Requirements", list())
         if not isinstance(requirements, list):
             requirements = [requirements]
         for requirement_yaml in requirements:
@@ -244,9 +244,7 @@ class Registry:
 
         # load protocols of each database
         for db_name, db_entries in protocols.items():
-            self._load_protocols(
-                db_name, db_entries, database_yml, mode=mode
-            )
+            self._load_protocols(db_name, db_entries, database_yml, mode=mode)
 
         # process "Databases" section
         databases = config.get("Databases", dict())
@@ -265,7 +263,6 @@ class Registry:
         # save configuration for later reloading of meta-protocols
         self.configs[database_yml] = config
 
-
     def get_database(self, database_name, **kwargs) -> Database:
         """Get database by name
 
@@ -284,7 +281,6 @@ class Registry:
             database = self.databases[database_name]
 
         except KeyError:
-
             if database_name == "X":
                 msg = (
                     "Could not find any meta-protocol. Please refer to "
@@ -302,7 +298,9 @@ class Registry:
 
         return database(**kwargs)
 
-    def get_protocol(self, name, preprocessors: Optional[Preprocessors] = None) -> Protocol:
+    def get_protocol(
+        self, name, preprocessors: Optional[Preprocessors] = None
+    ) -> Protocol:
         """Get protocol by full name
 
         Parameters
@@ -329,6 +327,14 @@ class Registry:
         protocol.name = name
         return protocol
 
+    # iterate over all protocols by name
+    def __iter__(self):
+        for database_name in self.databases:
+            database = self.get_database(database_name)
+            for task_name in database.get_tasks():
+                for protocol_name in database.get_protocols(task_name):
+                    yield f"{database_name}.{task_name}.{protocol_name}"
+
     def _load_protocols(
         self,
         db_name,
@@ -367,7 +373,9 @@ class Registry:
         # If needed, merge old protocols dict with the new one (according to current override rules)
         if db_name in self.databases:
             old_protocols = self.databases[db_name]._protocols
-            _merge_protocols_inplace(protocols, old_protocols, mode, db_name, database_yml)
+            _merge_protocols_inplace(
+                protocols, old_protocols, mode, db_name, database_yml
+            )
 
         # create database class on-the-fly
         protocol_list = [
@@ -389,13 +397,14 @@ class Registry:
         for db_yml, config in self.configs.items():
             databases = config.get("Protocols", dict())
             if "X" in databases:
-                self._load_protocols("X", databases["X"], db_yml, mode=LoadingMode.OVERRIDE)
-
+                self._load_protocols(
+                    "X", databases["X"], db_yml, mode=LoadingMode.OVERRIDE
+                )
 
 
 def _env_config_paths() -> List[Path]:
     """Parse PYANNOTE_DATABASE_CONFIG environment variable
-    
+
     PYANNOTE_DATABASE_CONFIG may contain multiple paths separation by ";".
 
     Returns
@@ -413,10 +422,11 @@ def _env_config_paths() -> List[Path]:
             paths.append(path)
     return paths
 
+
 def _find_default_ymls() -> List[Path]:
     """Get paths to default YAML configuration files
 
-    * $HOME/.pyannote/database.yml 
+    * $HOME/.pyannote/database.yml
     * $CWD/database.yml
     * PYANNOTE_DATABASE_CONFIG environment variable
 
@@ -431,21 +441,23 @@ def _find_default_ymls() -> List[Path]:
     home_db_yml = Path("~/.pyannote/database.yml").expanduser()
     if home_db_yml.is_file():
         paths.append(home_db_yml)
-    
+
     cwd_db_yml = Path.cwd() / "database.yml"
     if cwd_db_yml.is_file():
         paths.append(cwd_db_yml)
-    
+
     paths += _env_config_paths()
 
     return paths
 
+
 def _merge_protocols_inplace(
-    new_protocols: Dict[Tuple[Text, Text], Type], 
-    old_protocols: Dict[Tuple[Text, Text], Type], 
-    mode: LoadingMode, 
-    db_name: str, 
-    database_yml: str):
+    new_protocols: Dict[Tuple[Text, Text], Type],
+    old_protocols: Dict[Tuple[Text, Text], Type],
+    mode: LoadingMode,
+    db_name: str,
+    database_yml: str,
+):
     """Merge new and old protocols inplace into the passed new_protocol.
 
     Warning, merging order might be counterintuitive : "KEEP" strategy keeps element from the OLD protocol
@@ -471,7 +483,6 @@ def _merge_protocols_inplace(
 
     # for all previously defined protocol (in old_protocols)
     for p_id, old_p in old_protocols.items():
-        
         # if this protocol is redefined
         if p_id in new_protocols:
             t_name, p_name = p_id
@@ -480,13 +491,16 @@ def _merge_protocols_inplace(
             # raise an error
             if mode == LoadingMode.ERROR:
                 raise RuntimeError(
-                    f"Cannot load {realname} protocol from '{database_yml}' as it already exists.")
+                    f"Cannot load {realname} protocol from '{database_yml}' as it already exists."
+                )
 
             # keep the new protocol
             elif mode == LoadingMode.OVERRIDE:
-                warnings.warn(f"Replacing existing {realname} protocol by the one defined in '{database_yml}'.")
+                warnings.warn(
+                    f"Replacing existing {realname} protocol by the one defined in '{database_yml}'."
+                )
                 pass
-            
+
             # keep the old protocol
             elif mode == LoadingMode.KEEP:
                 warnings.warn(
@@ -498,9 +512,10 @@ def _merge_protocols_inplace(
         else:
             new_protocols[p_id] = old_p
 
+
 # initialize the registry singleton
 registry = Registry()
 
 # load all database yaml files found at startup
 for yml in _find_default_ymls():
-    registry.load_database(yml)
+    registry.load_database(yml)

{pyannote.database-5.0.0 → pyannote.database-5.1.0/pyannote.database.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyannote.database
-Version: 5.0.0
+Version: 5.1.0
 Summary: Interface to multimedia databases and experimental protocols
 Home-page: http://pyannote.github.io/
 Author: Hervé Bredin
@@ -9,10 +9,9 @@ Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Science/Research
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Natural Language :: English
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.6
-Classifier: Programming Language :: Python :: 3.7
-Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
 Classifier: Topic :: Scientific/Engineering
 Provides-Extra: testing
 Provides-Extra: doc

{pyannote.database-5.0.0 → pyannote.database-5.1.0}/pyannote.database.egg-info/requires.txt

@@ -1,7 +1,7 @@
 pyannote.core>=4.1
 pyYAML>=3.12
 pandas>=0.19
-typer[all]>=0.2.1
+typer>=0.12.1
 
 [:python_version < "3.8"]
 typing_extensions>=3.7.4

{pyannote.database-5.0.0 → pyannote.database-5.1.0}/setup.py

@@ -37,11 +37,13 @@ setup(
         "pyannote.core >= 4.1",
         "pyYAML >= 3.12",
         "pandas >= 0.19",
-        "typer[all] >= 0.2.1",
+        "typer >= 0.12.1",
         "typing_extensions >= 3.7.4;python_version < '3.8'",
     ],
     entry_points={
-        "console_scripts": ["pyannote-database=pyannote.database.cli:main",],
+        "console_scripts": [
+            "pyannote-database=pyannote.database.cli:main",
+        ],
         "pyannote.database.loader": [
             ".rttm = pyannote.database.loader:RTTMLoader",
             ".uem = pyannote.database.loader:UEMLoader",
@@ -65,10 +67,9 @@ setup(
         "Intended Audience :: Science/Research",
         "License :: OSI Approved :: MIT License",
         "Natural Language :: English",
-        "Programming Language :: Python :: 3",
-        "Programming Language :: Python :: 3.6",
-        "Programming Language :: Python :: 3.7",
-        "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
         "Topic :: Scientific/Engineering",
     ],
     extras_require={

{pyannote.database-5.0.0 → pyannote.database-5.1.0}/versioneer.py

@@ -1,4 +1,3 @@
-
 # Version: 0.15
 
 """
@@ -340,6 +339,7 @@ domain.
 """
 
 from __future__ import print_function
+
 try:
     import configparser
 except ImportError:
@@ -368,11 +368,13 @@ def get_root():
         setup_py = os.path.join(root, "setup.py")
         versioneer_py = os.path.join(root, "versioneer.py")
     if not (os.path.exists(setup_py) or os.path.exists(versioneer_py)):
-        err = ("Versioneer was unable to run the project root directory. "
-               "Versioneer requires setup.py to be executed from "
-               "its immediate directory (like 'python setup.py COMMAND'), "
-               "or in a way that lets it use sys.argv[0] to find the root "
-               "(like 'python path/to/setup.py COMMAND').")
+        err = (
+            "Versioneer was unable to run the project root directory. "
+            "Versioneer requires setup.py to be executed from "
+            "its immediate directory (like 'python setup.py COMMAND'), "
+            "or in a way that lets it use sys.argv[0] to find the root "
+            "(like 'python path/to/setup.py COMMAND')."
+        )
         raise VersioneerBadRootError(err)
     try:
         # Certain runtime workflows (setup.py install/develop in a setuptools
@@ -383,8 +385,10 @@ def get_root():
         # versioneer.py was first imported, even in later projects.
         me = os.path.realpath(os.path.abspath(__file__))
         if os.path.splitext(me)[0] != os.path.splitext(versioneer_py)[0]:
-            print("Warning: build in %s is using versioneer.py from %s"
-                  % (os.path.dirname(me), versioneer_py))
+            print(
+                "Warning: build in %s is using versioneer.py from %s"
+                % (os.path.dirname(me), versioneer_py)
+            )
     except NameError:
         pass
     return root
@@ -396,15 +400,16 @@ def get_config_from_root(root):
     # configparser.NoOptionError (if it lacks "VCS="). See the docstring at
     # the top of versioneer.py for instructions on writing your setup.cfg .
     setup_cfg = os.path.join(root, "setup.cfg")
-    parser = configparser.SafeConfigParser()
+    parser = configparser.ConfigParser()
     with open(setup_cfg, "r") as f:
-        parser.readfp(f)
+        parser.read_file(f)
     VCS = parser.get("versioneer", "VCS")  # mandatory
 
     def get(parser, name):
         if parser.has_option("versioneer", name):
             return parser.get("versioneer", name)
         return None
+
     cfg = VersioneerConfig()
     cfg.VCS = VCS
     cfg.style = get(parser, "style") or ""
@@ -419,6 +424,7 @@ def get_config_from_root(root):
 class NotThisMethod(Exception):
     pass
 
+
 # these dictionaries contain VCS-specific tools
 LONG_VERSION_PY = {}
 HANDLERS = {}
@@ -430,6 +436,7 @@ def register_vcs_handler(vcs, method):  # decorator
             HANDLERS[vcs] = {}
         HANDLERS[vcs][method] = f
         return f
+
     return decorate
 
 
@@ -440,9 +447,12 @@ def run_command(commands, args, cwd=None, verbose=False, hide_stderr=False):
         try:
             dispcmd = str([c] + args)
             # remember shell=False, so use git.cmd on windows, not just git
-            p = subprocess.Popen([c] + args, cwd=cwd, stdout=subprocess.PIPE,
-                                 stderr=(subprocess.PIPE if hide_stderr
-                                         else None))
+            p = subprocess.Popen(
+                [c] + args,
+                cwd=cwd,
+                stdout=subprocess.PIPE,
+                stderr=(subprocess.PIPE if hide_stderr else None),
+            )
             break
         except EnvironmentError:
             e = sys.exc_info()[1]
@@ -464,7 +474,11 @@ def run_command(commands, args, cwd=None, verbose=False, hide_stderr=False):
             print("unable to run %s (error)" % dispcmd)
         return None
     return stdout
-LONG_VERSION_PY['git'] = '''
+
+
+LONG_VERSION_PY[
+    "git"
+] = """
 # This file helps to compute a version number in source trees obtained from
 # git-archive tarball (such as those provided by githubs download-from-tag
 # feature). Distribution tarballs (built by setup.py sdist) and build
@@ -924,7 +938,7 @@ def get_versions():
     return {"version": "0+unknown", "full-revisionid": None,
             "dirty": None,
             "error": "unable to compute version"}
-'''
+"""
 
 
 @register_vcs_handler("git", "get_keywords")
@@ -964,7 +978,7 @@ def git_versions_from_keywords(keywords, tag_prefix, verbose):
     # starting in git-1.8.3, tags are listed as "tag: foo-1.0" instead of
     # just "foo-1.0". If we see a "tag: " prefix, prefer those.
     TAG = "tag: "
-    tags = set([r[len(TAG):] for r in refs if r.startswith(TAG)])
+    tags = set([r[len(TAG) :] for r in refs if r.startswith(TAG)])
     if not tags:
         # Either we're using git < 1.8.3, or there really are no tags. We use
         # a heuristic: assume all version tags have a digit. The old git %d
@@ -973,27 +987,32 @@ def git_versions_from_keywords(keywords, tag_prefix, verbose):
         # between branches and tags. By ignoring refnames without digits, we
         # filter out many common branch names like "release" and
         # "stabilization", as well as "HEAD" and "master".
-        tags = set([r for r in refs if re.search(r'\d', r)])
+        tags = set([r for r in refs if re.search(r"\d", r)])
         if verbose:
-            print("discarding '%s', no digits" % ",".join(refs-tags))
+            print("discarding '%s', no digits" % ",".join(refs - tags))
     if verbose:
         print("likely tags: %s" % ",".join(sorted(tags)))
     for ref in sorted(tags):
         # sorting will prefer e.g. "2.0" over "2.0rc1"
         if ref.startswith(tag_prefix):
-            r = ref[len(tag_prefix):]
+            r = ref[len(tag_prefix) :]
             if verbose:
                 print("picking %s" % r)
-            return {"version": r,
-                    "full-revisionid": keywords["full"].strip(),
-                    "dirty": False, "error": None
-                    }
+            return {
+                "version": r,
+                "full-revisionid": keywords["full"].strip(),
+                "dirty": False,
+                "error": None,
+            }
     # no suitable tags, so version is "0+unknown", but full hex is still there
     if verbose:
         print("no suitable tags, using unknown + full revision id")
-    return {"version": "0+unknown",
-            "full-revisionid": keywords["full"].strip(),
-            "dirty": False, "error": "no suitable tags"}
+    return {
+        "version": "0+unknown",
+        "full-revisionid": keywords["full"].strip(),
+        "dirty": False,
+        "error": "no suitable tags",
+    }
 
 
 @register_vcs_handler("git", "pieces_from_vcs")
@@ -1013,9 +1032,9 @@ def git_pieces_from_vcs(tag_prefix, root, verbose, run_command=run_command):
         GITS = ["git.cmd", "git.exe"]
     # if there is a tag, this yields TAG-NUM-gHEX[-dirty]
     # if there are no tags, this yields HEX[-dirty] (no NUM)
-    describe_out = run_command(GITS, ["describe", "--tags", "--dirty",
-                                      "--always", "--long"],
-                               cwd=root)
+    describe_out = run_command(
+        GITS, ["describe", "--tags", "--dirty", "--always", "--long"], cwd=root
+    )
     # --long was added in git-1.5.5
     if describe_out is None:
         raise NotThisMethod("'git describe' failed")
@@ -1038,17 +1057,16 @@ def git_pieces_from_vcs(tag_prefix, root, verbose, run_command=run_command):
     dirty = git_describe.endswith("-dirty")
     pieces["dirty"] = dirty
     if dirty:
-        git_describe = git_describe[:git_describe.rindex("-dirty")]
+        git_describe = git_describe[: git_describe.rindex("-dirty")]
 
     # now we have TAG-NUM-gHEX or HEX
 
     if "-" in git_describe:
         # TAG-NUM-gHEX
-        mo = re.search(r'^(.+)-(\d+)-g([0-9a-f]+)$', git_describe)
+        mo = re.search(r"^(.+)-(\d+)-g([0-9a-f]+)$", git_describe)
         if not mo:
             # unparseable. Maybe git-describe is misbehaving?
-            pieces["error"] = ("unable to parse git-describe output: '%s'"
-                               % describe_out)
+            pieces["error"] = "unable to parse git-describe output: '%s'" % describe_out
             return pieces
 
         # tag
@@ -1057,10 +1075,12 @@ def git_pieces_from_vcs(tag_prefix, root, verbose, run_command=run_command):
             if verbose:
                 fmt = "tag '%s' doesn't start with prefix '%s'"
                 print(fmt % (full_tag, tag_prefix))
-            pieces["error"] = ("tag '%s' doesn't start with prefix '%s'"
-                               % (full_tag, tag_prefix))
+            pieces["error"] = "tag '%s' doesn't start with prefix '%s'" % (
+                full_tag,
+                tag_prefix,
+            )
             return pieces
-        pieces["closest-tag"] = full_tag[len(tag_prefix):]
+        pieces["closest-tag"] = full_tag[len(tag_prefix) :]
 
         # distance: number of commits since tag
         pieces["distance"] = int(mo.group(2))
@@ -1071,8 +1091,7 @@ 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
@@ -1117,12 +1136,18 @@ def versions_from_parentdir(parentdir_prefix, root, verbose):
     dirname = os.path.basename(root)
     if not dirname.startswith(parentdir_prefix):
         if verbose:
-            print("guessing rootdir is '%s', but '%s' doesn't start with "
-                  "prefix '%s'" % (root, dirname, parentdir_prefix))
+            print(
+                "guessing rootdir is '%s', but '%s' doesn't start with "
+                "prefix '%s'" % (root, dirname, parentdir_prefix)
+            )
         raise NotThisMethod("rootdir doesn't start with parentdir_prefix")
-    return {"version": dirname[len(parentdir_prefix):],
-            "full-revisionid": None,
-            "dirty": False, "error": None}
+    return {
+        "version": dirname[len(parentdir_prefix) :],
+        "full-revisionid": None,
+        "dirty": False,
+        "error": None,
+    }
+
 
 SHORT_VERSION_PY = """
 # This file was generated by 'versioneer.py' (0.15) from
@@ -1149,8 +1174,9 @@ def versions_from_file(filename):
             contents = f.read()
     except EnvironmentError:
         raise NotThisMethod("unable to read _version.py")
-    mo = re.search(r"version_json = '''\n(.*)'''  # END VERSION_JSON",
-                   contents, re.M | re.S)
+    mo = re.search(
+        r"version_json = '''\n(.*)'''  # END VERSION_JSON", contents, re.M | re.S
+    )
     if not mo:
         raise NotThisMethod("no version_json in _version.py")
     return json.loads(mo.group(1))
@@ -1158,8 +1184,7 @@ def versions_from_file(filename):
 
 def write_to_version_file(filename, versions):
     os.unlink(filename)
-    contents = json.dumps(versions, sort_keys=True,
-                          indent=1, separators=(",", ": "))
+    contents = json.dumps(versions, sort_keys=True, indent=1, separators=(",", ": "))
     with open(filename, "w") as f:
         f.write(SHORT_VERSION_PY % contents)
 
@@ -1189,8 +1214,7 @@ def render_pep440(pieces):
                 rendered += ".dirty"
     else:
         # exception #1
-        rendered = "0+untagged.%d.g%s" % (pieces["distance"],
-                                          pieces["short"])
+        rendered = "0+untagged.%d.g%s" % (pieces["distance"], pieces["short"])
         if pieces["dirty"]:
             rendered += ".dirty"
     return rendered
@@ -1297,10 +1321,12 @@ def render_git_describe_long(pieces):
 
 def render(pieces, style):
     if pieces["error"]:
-        return {"version": "unknown",
-                "full-revisionid": pieces.get("long"),
-                "dirty": None,
-                "error": pieces["error"]}
+        return {
+            "version": "unknown",
+            "full-revisionid": pieces.get("long"),
+            "dirty": None,
+            "error": pieces["error"],
+        }
 
     if not style or style == "default":
         style = "pep440"  # the default
@@ -1320,8 +1346,12 @@ def render(pieces, style):
     else:
         raise ValueError("unknown style '%s'" % style)
 
-    return {"version": rendered, "full-revisionid": pieces["long"],
-            "dirty": pieces["dirty"], "error": None}
+    return {
+        "version": rendered,
+        "full-revisionid": pieces["long"],
+        "dirty": pieces["dirty"],
+        "error": None,
+    }
 
 
 class VersioneerBadRootError(Exception):
@@ -1342,8 +1372,9 @@ def get_versions(verbose=False):
     handlers = HANDLERS.get(cfg.VCS)
     assert handlers, "unrecognized VCS '%s'" % cfg.VCS
     verbose = verbose or cfg.verbose
-    assert cfg.versionfile_source is not None, \
-        "please set versioneer.versionfile_source"
+    assert (
+        cfg.versionfile_source is not None
+    ), "please set versioneer.versionfile_source"
     assert cfg.tag_prefix is not None, "please set versioneer.tag_prefix"
 
     versionfile_abs = os.path.join(root, cfg.versionfile_source)
@@ -1397,8 +1428,12 @@ def get_versions(verbose=False):
     if verbose:
         print("unable to compute version")
 
-    return {"version": "0+unknown", "full-revisionid": None,
-            "dirty": None, "error": "unable to compute version"}
+    return {
+        "version": "0+unknown",
+        "full-revisionid": None,
+        "dirty": None,
+        "error": "unable to compute version",
+    }
 
 
 def get_version():
@@ -1444,6 +1479,7 @@ def get_cmdclass():
             print(" dirty: %s" % vers.get("dirty"))
             if vers["error"]:
                 print(" error: %s" % vers["error"])
+
     cmds["version"] = cmd_version
 
     # we override "build_py" in both distutils and setuptools
@@ -1467,10 +1503,10 @@ def get_cmdclass():
             # now locate _version.py in the new build/ directory and replace
             # it with an updated value
             if cfg.versionfile_build:
-                target_versionfile = os.path.join(self.build_lib,
-                                                  cfg.versionfile_build)
+                target_versionfile = os.path.join(self.build_lib, cfg.versionfile_build)
                 print("UPDATING %s" % target_versionfile)
                 write_to_version_file(target_versionfile, versions)
+
     cmds["build_py"] = cmd_build_py
 
     if "cx_Freeze" in sys.modules:  # cx_freeze enabled?
@@ -1489,13 +1525,17 @@ def get_cmdclass():
                 os.unlink(target_versionfile)
                 with open(cfg.versionfile_source, "w") as f:
                     LONG = LONG_VERSION_PY[cfg.VCS]
-                    f.write(LONG %
-                            {"DOLLAR": "$",
-                             "STYLE": cfg.style,
-                             "TAG_PREFIX": cfg.tag_prefix,
-                             "PARENTDIR_PREFIX": cfg.parentdir_prefix,
-                             "VERSIONFILE_SOURCE": cfg.versionfile_source,
-                             })
+                    f.write(
+                        LONG
+                        % {
+                            "DOLLAR": "$",
+                            "STYLE": cfg.style,
+                            "TAG_PREFIX": cfg.tag_prefix,
+                            "PARENTDIR_PREFIX": cfg.parentdir_prefix,
+                            "VERSIONFILE_SOURCE": cfg.versionfile_source,
+                        }
+                    )
+
         cmds["build_exe"] = cmd_build_exe
         del cmds["build_py"]
 
@@ -1523,8 +1563,10 @@ def get_cmdclass():
             # updated value
             target_versionfile = os.path.join(base_dir, cfg.versionfile_source)
             print("UPDATING %s" % target_versionfile)
-            write_to_version_file(target_versionfile,
-                                  self._versioneer_generated_versions)
+            write_to_version_file(
+                target_versionfile, self._versioneer_generated_versions
+            )
+
     cmds["sdist"] = cmd_sdist
 
     return cmds
@@ -1578,11 +1620,13 @@ def do_setup():
     root = get_root()
     try:
         cfg = get_config_from_root(root)
-    except (EnvironmentError, configparser.NoSectionError,
-            configparser.NoOptionError) as e:
+    except (
+        EnvironmentError,
+        configparser.NoSectionError,
+        configparser.NoOptionError,
+    ) as e:
         if isinstance(e, (EnvironmentError, configparser.NoSectionError)):
-            print("Adding sample versioneer config to setup.cfg",
-                  file=sys.stderr)
+            print("Adding sample versioneer config to setup.cfg", file=sys.stderr)
             with open(os.path.join(root, "setup.cfg"), "a") as f:
                 f.write(SAMPLE_CONFIG)
         print(CONFIG_ERROR, file=sys.stderr)
@@ -1591,15 +1635,18 @@ def do_setup():
     print(" creating %s" % cfg.versionfile_source)
     with open(cfg.versionfile_source, "w") as f:
         LONG = LONG_VERSION_PY[cfg.VCS]
-        f.write(LONG % {"DOLLAR": "$",
-                        "STYLE": cfg.style,
-                        "TAG_PREFIX": cfg.tag_prefix,
-                        "PARENTDIR_PREFIX": cfg.parentdir_prefix,
-                        "VERSIONFILE_SOURCE": cfg.versionfile_source,
-                        })
-
-    ipy = os.path.join(os.path.dirname(cfg.versionfile_source),
-                       "__init__.py")
+        f.write(
+            LONG
+            % {
+                "DOLLAR": "$",
+                "STYLE": cfg.style,
+                "TAG_PREFIX": cfg.tag_prefix,
+                "PARENTDIR_PREFIX": cfg.parentdir_prefix,
+                "VERSIONFILE_SOURCE": cfg.versionfile_source,
+            }
+        )
+
+    ipy = os.path.join(os.path.dirname(cfg.versionfile_source), "__init__.py")
     if os.path.exists(ipy):
         try:
             with open(ipy, "r") as f:
@@ -1641,8 +1688,10 @@ def do_setup():
     else:
         print(" 'versioneer.py' already in MANIFEST.in")
     if cfg.versionfile_source not in simple_includes:
-        print(" appending versionfile_source ('%s') to MANIFEST.in" %
-              cfg.versionfile_source)
+        print(
+            " appending versionfile_source ('%s') to MANIFEST.in"
+            % cfg.versionfile_source
+        )
         with open(manifest_in, "a") as f:
             f.write("include %s\n" % cfg.versionfile_source)
     else:
@@ -1690,6 +1739,7 @@ def scan_setup_py():
         errors += 1
     return errors
 
+
 if __name__ == "__main__":
     cmd = sys.argv[1]
     if cmd == "setup":