diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 00000000..cc298b01 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,19 @@ +###### Requirements without Version Specifiers ###### +iminuit +lightgbm +scikit-learn +seaborn +setuptools_scm + +###### Requirements with Version Specifiers ###### +astropy>=1.1.2 +future>=0.16 +george>=0.3.0 +matplotlib>=1.5.1 +nose>=1.3.7 +numpy>=1.18.4 +pandas>=0.23.0 +pyyaml>=3.13 +pywavelets>=0.4.0 +scipy>=0.17.0 +sncosmo==2.1.0 \ No newline at end of file diff --git a/setup.py b/setup.py index 03ad42dc..59b0f85a 100644 --- a/setup.py +++ b/setup.py @@ -38,6 +38,12 @@ def run(self): setup( name='snmachine', + author='Michelle Lochner', + author_email='dr.michelle.lochner@gmail.com', + description='Machine learning code for photometric supernova ' + 'classification', + url='https://github.com/LSSTDESC/snmachine', + license='BSD-3-Clause License', use_scm_version={ "root": ".", "relative_to": __file__, @@ -50,10 +56,20 @@ def run(self): 'example_data/example_data_for_tests.pckl']}, exclude_package_data={'utils': ['archive/*']}, cmdclass={'install': ExtractExampleData}, - url='https://github.com/LSSTDESC/snmachine', - license='BSD-3-Clause License', - author='Michelle Lochner', - author_email='dr.michelle.lochner@gmail.com', - description='Machine learning code for photometric supernova ' - 'classification' + install_requires=['astropy>=1.1.2', + 'matplotlib>=1.5.1', + 'numpy>=1.18.4', + 'scikit-learn', + 'scipy>=0.17.0', + 'george>=0.3.0', + 'iminuit', + 'pandas>=0.23.0', + 'pywavelets>=0.4.0', + 'sncosmo==2.1.0', + 'nose>=1.3.7', + 'future>=0.16', + 'pyyaml>=3.13', + 'seaborn', + 'lightgbm', + 'setuptools_scm'] )