diff --git a/setup.py b/setup.py index 55f5dd7..360d5de 100644 --- a/setup.py +++ b/setup.py @@ -8,14 +8,14 @@ setuptools.setup( name="tensordiffeq", - version="0.1.4", + version="0.1.5", author="Levi McClenny", author_email="levimcclenny@tamu.edu", description="Distributed PDE Solver in Tensorflow", long_description=long_description, long_description_content_type="text/markdown", url="https://github.com/tensordiffeq/tensordiffeq", - download_url="https://github.com/tensordiffeq/tensordiffeq/tarball/v0.1.4", + download_url="https://github.com/tensordiffeq/tensordiffeq/tarball/v0.1.5", install_requires=install_requires, packages=setuptools.find_packages(), include_package_data=True, diff --git a/tensordiffeq/__init__.py b/tensordiffeq/__init__.py new file mode 100644 index 0000000..d2eef1f --- /dev/null +++ b/tensordiffeq/__init__.py @@ -0,0 +1,23 @@ +from __future__ import absolute_import + +from tensordiffeq import models, optimizers, networks, plotting, utils, domains, boundaries, fit, helpers + +# from .models import CollocationSolverND, DiscoveryModel +# from .boundaries import dirichletBC, periodicBC, IC +# from .utils import constant, LatinHypercubeSample, tensor +# from .plotting import newfig, get_griddata +# from .helpers import find_L2_error +# from .optimizers import graph_lbfgs, eager_lbfgs + + +__all__ = [ + "models", + "networks", + "plotting", + "utils", + "helpers", + "optimizers", + "boundaries", + "domains", + "fit", +]