Run black and isort and add to CI

.github/workflows/build.yml

@@ -34,4 +34,9 @@ jobs:
           make html -C docs/ SPHINXOPTS="-W --keep-going -n"
       - name: Doctest
         run: |
-          make doctest -C docs/
+          make doctest -C docs/
+      - name: Lint
+        run: |
+          pip install black==22.1.0 isort
+          black . --check --diff
+          isort . --check --verbose --only-modified --diff

docs/source/conf.py

@@ -10,28 +10,29 @@
 # All configuration values have a default; values that are commented out
 # serve to show the default.
 
+import sys
+
 # If extensions (or modules to document with autodoc) are in another directory,
 # add these directories to sys.path here. If the directory is relative to the
 # documentation root, use os.path.abspath to make it absolute, like shown here.
 from pathlib import Path
-import sys
 
 from docutils.nodes import Text
 from sphinx.ext.intersphinx import missing_reference
 
-
 import gbd_mapping
+
 base_dir = Path(gbd_mapping.__file__).parent
 
 about = {}
 with (base_dir / "__about__.py").open() as f:
     exec(f.read(), about)
 
-sys.path.insert(0, str(Path('..').resolve()))
+sys.path.insert(0, str(Path("..").resolve()))
 
 # -- Project information -----------------------------------------------------
 
-project = about['__title__']
+project = about["__title__"]
 copyright = f'2022, {about["__author__"]}'
 author = about["__author__"]
 
@@ -45,33 +46,33 @@
 
 # If your documentation needs a minimal Sphinx version, state it here.
 
-needs_sphinx = '4.0'
+needs_sphinx = "4.0"
 
 # Add any Sphinx extension module names here, as strings. They can be
 # extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
 # ones.
 extensions = [
-    'sphinx.ext.autodoc',
-    'sphinx.ext.intersphinx',
-    'sphinx.ext.doctest',
-    'sphinx.ext.todo',
-    'sphinx.ext.coverage',
-    'sphinx.ext.mathjax',
-    'sphinx.ext.napoleon',
-    'sphinx.ext.viewcode',
+    "sphinx.ext.autodoc",
+    "sphinx.ext.intersphinx",
+    "sphinx.ext.doctest",
+    "sphinx.ext.todo",
+    "sphinx.ext.coverage",
+    "sphinx.ext.mathjax",
+    "sphinx.ext.napoleon",
+    "sphinx.ext.viewcode",
 ]
 
 # Add any paths that contain templates here, relative to this directory.
-templates_path = ['_templates']
+templates_path = ["_templates"]
 
 # The suffix(es) of source filenames.
 # You can specify multiple suffix as a list of string:
 #
 # source_suffix = ['.rst', '.md']
-source_suffix = '.rst'
+source_suffix = ".rst"
 
 # The master toctree document.
-master_doc = 'index'
+master_doc = "index"
 
 # General information about the project.
 
@@ -89,7 +90,7 @@
 exclude_patterns = []
 
 # The name of the Pygments (syntax highlighting) style to use.
-pygments_style = 'sphinx'
+pygments_style = "sphinx"
 
 # If true, `todo` and `todoList` produce output, else they produce nothing.
 todo_include_todos = True
@@ -100,8 +101,8 @@
 # The theme to use for HTML and HTML Help pages.  See the documentation for
 # a list of builtin themes.
 
-html_theme_path = ['_theme']
-html_theme = 'sphinx_rtd_theme'
+html_theme_path = ["_theme"]
+html_theme = "sphinx_rtd_theme"
 
 # Theme options are theme-specific and customize the look and feel of a theme
 # further.  For a list of options available for each theme, see the
@@ -112,17 +113,17 @@
 # Add any paths that contain custom static files (such as style sheets) here,
 # relative to this directory. They are copied after the builtin static files,
 # so a file named "default.css" will overwrite the builtin "default.css".
-html_static_path = ['_static']
+html_static_path = ["_static"]
 
 # Custom sidebar templates, must be a dictionary that maps document names
 # to template names.
 #
 # This is required for the alabaster theme
 # refs: http://alabaster.readthedocs.io/en/latest/installation.html#sidebars
 html_sidebars = {
-    '**': [
-        'globaltoc.html',  # needs 'show_related': True theme option to display
-        'searchbox.html',
+    "**": [
+        "globaltoc.html",  # needs 'show_related': True theme option to display
+        "searchbox.html",
     ]
 }
 
@@ -139,15 +140,12 @@
     # The paper size ('letterpaper' or 'a4paper').
     #
     # 'papersize': 'letterpaper',
-
     # The font size ('10pt', '11pt' or '12pt').
     #
     # 'pointsize': '10pt',
-
     # Additional stuff for the LaTeX preamble.
     #
     # 'preamble': '',
-
     # Latex figure (float) alignment
     #
     # 'figure_align': 'htbp',
@@ -157,8 +155,13 @@
 # (source start file, target name, title,
 #  author, documentclass [howto, manual, or own class]).
 latex_documents = [
-    (master_doc, f'{about["__title__"]}.tex', f'{about["__title__"]} Documentation',
-     about["__author__"], 'manual'),
+    (
+        master_doc,
+        f'{about["__title__"]}.tex',
+        f'{about["__title__"]} Documentation',
+        about["__author__"],
+        "manual",
+    ),
 ]
 
 
@@ -167,8 +170,7 @@
 # One entry per manual page. List of tuples
 # (source start file, name, description, authors, manual section).
 man_pages = [
-    (master_doc, f'{about["__title__"]}', f'{about["__title__"]} Documentation',
-     [author], 1)
+    (master_doc, f'{about["__title__"]}', f'{about["__title__"]} Documentation', [author], 1)
 ]
 
 
@@ -178,35 +180,43 @@
 # (source start file, target name, title, author,
 #  dir menu entry, description, category)
 texinfo_documents = [
-    (master_doc, f'{about["__title__"]}', f'{about["__title__"]} Documentation',
-     author, f'{about["__title__"]}', about["__summary__"],
-     'Miscellaneous'),
+    (
+        master_doc,
+        f'{about["__title__"]}',
+        f'{about["__title__"]} Documentation',
+        author,
+        f'{about["__title__"]}',
+        about["__summary__"],
+        "Miscellaneous",
+    ),
 ]
 
 # Other docs we can link to
-intersphinx_mapping = {'python': ('https://docs.python.org/3.8', None),
-                       'pandas': ('https://pandas.pydata.org/pandas-docs/stable/', None),
-                       'tables': ('https://www.pytables.org/', None),
-                       'numpy': ('https://numpy.org/doc/stable/', None),
-                       'networkx': ('https://networkx.org/documentation/stable/', None)}
+intersphinx_mapping = {
+    "python": ("https://docs.python.org/3.8", None),
+    "pandas": ("https://pandas.pydata.org/pandas-docs/stable/", None),
+    "tables": ("https://www.pytables.org/", None),
+    "numpy": ("https://numpy.org/doc/stable/", None),
+    "networkx": ("https://networkx.org/documentation/stable/", None),
+}
 
 
 # -- Autodoc configuration ------------------------------------------------
 
 autodoc_default_options = {
     # Automatically document members (e.g. classes in a module,
     # methods in a class, etc.)
-    'members': True,
+    "members": True,
     # Order of items documented is determined by the order
     # of appearance in the source code
-    'member-order': 'bysource',
+    "member-order": "bysource",
     # Generate docs even if an item has no docstring.
-    'undoc-members': True,
+    "undoc-members": True,
     # Don't document things with a leading underscore.
-    'private-members': False,
+    "private-members": False,
 }
 # Display type hints in the description instead of the signature.
-autodoc_typehints = 'description'
+autodoc_typehints = "description"
 
 
 # -- nitpicky mode --------------------------------------------------------
@@ -215,7 +225,7 @@
 nitpicky = True
 
 nitpick_ignore = []
-for line in open('../nitpick-exceptions'):
+for line in open("../nitpick-exceptions"):
     if line.strip() == "" or line.startswith("#"):
         continue
     dtype, target = line.split(None, 1)

pyproject.toml

@@ -0,0 +1,13 @@
+[tool.black]
+line_length = 94
+exclude = '''
+/(
+   build
+   | src/gbd_mapping
+)/
+
+'''
+
+[tool.isort]
+profile = "black"
+skip = ["build", "src/gbd_mapping"]

setup.py

@@ -1,8 +1,7 @@
 #!/usr/bin/env python
 import os
 
-from setuptools import setup, find_packages
-
+from setuptools import find_packages, setup
 
 if __name__ == "__main__":
 
@@ -17,54 +16,48 @@
         long_description = f.read()
 
     install_requirements = [
-        'click',
-        'numpy',
-        'pandas',
-        'pyyaml',
+        "click",
+        "numpy",
+        "pandas",
+        "pyyaml",
     ]
 
     data_requires = [
-        'vivarium-gbd-access>=3.0.2',
+        "vivarium-gbd-access>=3.0.2",
     ]
 
     test_requirements = [
-        'pytest',
-        'pytest-mock',
+        "pytest",
+        "pytest-mock",
     ]
 
     doc_requirements = [
-        'sphinx',
-        'sphinx-rtd-theme',
+        "sphinx",
+        "sphinx-rtd-theme",
     ]
 
     setup(
-        name=about['__title__'],
-        version=about['__version__'],
-
-        description=about['__summary__'],
+        name=about["__title__"],
+        version=about["__version__"],
+        description=about["__summary__"],
         long_description=long_description,
         url=about["__uri__"],
-
         author=about["__author__"],
         author_email=about["__email__"],
-
-        package_dir={'': 'src'},
-        packages=find_packages(where='src'),
+        package_dir={"": "src"},
+        packages=find_packages(where="src"),
         include_package_data=True,
-
         install_requires=install_requirements,
         tests_require=test_requirements,
         extras_require={
-            'docs': doc_requirements,
-            'test': test_requirements,
-            'data': data_requires,
-            'dev': doc_requirements + test_requirements + data_requires,
+            "docs": doc_requirements,
+            "test": test_requirements,
+            "data": data_requires,
+            "dev": doc_requirements + test_requirements + data_requires,
         },
-
         entry_points="""
                 [console_scripts]
                 build_mapping=gbd_mapping_generator.build_mapping:build_mapping
             """,
-
         zip_safe=False,
     )