diff --git a/__version__.py b/__version__.py deleted file mode 100644 index 4609a4128406cca44479c45c79cacbea5d94badb..0000000000000000000000000000000000000000 --- a/__version__.py +++ /dev/null @@ -1,14 +0,0 @@ -# coding: utf-8 - -""" -TensorFlow implementation of the Lorentz Boost Network (LBN). https://arxiv.org/abs/1812.09722. -""" - - -__author__ = "Marcel Rieger" -__copyright__ = "Copyright 2018-2019, Marcel Rieger" -__license__ = "BSD" -__credits__ = ["Martin Erdmann", "Erik Geiser", "Yannik Rath", "Marcel Rieger"] -__contact__ = "https://git.rwth-aachen.de/3pia/lbn" -__email__ = "marcel.rieger@cern.ch" -__version__ = "1.0.1" diff --git a/lbn.py b/lbn.py index 9c866d03b920bdbc46a64425257e86a4d1c83a2b..99cebbf7048451601f5778e2fea521a357813bd9 100644 --- a/lbn.py +++ b/lbn.py @@ -5,6 +5,14 @@ TensorFlow implementation of the Lorentz Boost Network (LBN). https://arxiv.org/ """ +__author__ = "Marcel Rieger" +__copyright__ = "Copyright 2018-2019, Marcel Rieger" +__license__ = "BSD" +__credits__ = ["Martin Erdmann", "Erik Geiser", "Yannik Rath", "Marcel Rieger"] +__contact__ = "https://git.rwth-aachen.de/3pia/lbn" +__email__ = "marcel.rieger@cern.ch" +__version__ = "1.0.1" + __all__ = ["LBN", "FeatureFactoryBase", "FeatureFactory"] diff --git a/requirements.txt b/requirements.txt index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0c8c3d2e7969d38f8b76ec8f79e2067fea30cdaa 100644 --- a/requirements.txt +++ b/requirements.txt @@ -0,0 +1,2 @@ +numpy +tensorflow diff --git a/setup.py b/setup.py index 361dcc0f349ac3c88fc325516bcdf54aa10ba4b8..e7909bb6d3eddee5bceda35c4d950e532ff1c883 100644 --- a/setup.py +++ b/setup.py @@ -34,20 +34,14 @@ with open(os.path.join(this_dir, "README.md"), "r") as f: with open(os.path.join(this_dir, "requirements.txt"), "r") as f: install_requires = [line.strip() for line in f.readlines() if line.strip()] -# load package infos -pkg = {} -with open(os.path.join(this_dir, "__version__.py"), "r") as f: - exec(f.read(), pkg) - - setuptools.setup( - name="lbn", - version=pkg["__version__"], - author=pkg["__author__"], - author_email=pkg["__email__"], - description=pkg["__doc__"].strip().split("\n")[0].strip(), - license=pkg["__license__"], - url=pkg["__contact__"], + name=lbn.__name__, + version=lbn.__version__, + author=lbn.__author__, + author_email=lbn.__email__, + description=lbn.__doc__.strip().split("\n")[0].strip(), + license=lbn.__license__, + url=lbn.__contact__, keywords=keywords, classifiers=classifiers, long_description=long_description, @@ -55,6 +49,6 @@ setuptools.setup( install_requires=install_requires, python_requires=">=2.7", zip_safe=False, - py_modules=["lbn"], + py_modules=[lbn.__name__], data_files=[(".", ["LICENSE", "requirements.txt", "README.md"])], )