Merge pull request #26 from ihmeuw-msca/format

Format

Author: zhengp0

setup.py

@@ -3,47 +3,40 @@
 from setuptools import setup, find_packages
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     base_dir = Path(__file__).parent
-    src_dir = base_dir/'src'/'xspline'
+    src_dir = base_dir / "src" / "xspline"
 
     sys.path.insert(0, src_dir.as_posix())
     import __about__ as about
 
-    with (base_dir/'README.rst').open() as f:
+    with (base_dir / "README.rst").open() as f:
         long_description = f.read()
 
-    install_requirements = [
-        'numpy'
-    ]
+    install_requirements = ["numpy"]
 
-    test_requirements = [
-        'pytest',
-        'pytest-mock'
-    ]
+    test_requirements = ["pytest", "pytest-mock"]
 
     doc_requirements = []
 
-    setup(name=about.__title__,
-          version=about.__version__,
-
-          description=about.__summary__,
-          long_description=long_description,
-          license=about.__license__,
-          url=about.__uri__,
-
-          author=about.__author__,
-          author_email=about.__email__,
-
-          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,
-              'dev': doc_requirements + test_requirements
-          },
-          zip_safe=False,)
+    setup(
+        name=about.__title__,
+        version=about.__version__,
+        description=about.__summary__,
+        long_description=long_description,
+        license=about.__license__,
+        url=about.__uri__,
+        author=about.__author__,
+        author_email=about.__email__,
+        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,
+            "dev": doc_requirements + test_requirements,
+        },
+        zip_safe=False,
+    )

src/xspline/__about__.py

@@ -1,6 +1,12 @@
 __all__ = [
-    "__title__", "__summary__", "__uri__", "__version__", "__author__",
-    "__email__", "__license__", "__copyright__",
+    "__title__",
+    "__summary__",
+    "__uri__",
+    "__version__",
+    "__author__",
+    "__email__",
+    "__license__",
+    "__copyright__",
 ]
 
 __title__ = "xspline"

src/xspline/__init__.py

@@ -1,9 +1,19 @@
-# -*- coding: utf-8 -*-
-"""
-    xspline
-    ~~~~~~~
-
-    xspline package
-"""
-from .core import *
 from . import utils
+from .core import (
+    NDXSpline,
+    XSpline,
+    bspline_dfun,
+    bspline_domain,
+    bspline_fun,
+    bspline_ifun,
+)
+
+__all__ = [
+    "utils",
+    "bspline_domain",
+    "bspline_fun",
+    "bspline_dfun",
+    "bspline_ifun",
+    "XSpline",
+    "NDXSpline",
+]