-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
32 lines (29 loc) · 830 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import setuptools
with open("README.md", "r") as fh:
long_description = fh.read()
with open("requirements.txt", "r") as f:
requirements = f.readlines()
setuptools.setup(
name="MARS",
version="0.0.1",
author="Krunoslav Lehman Pavasovic, Jonas Rothfuss",
author_email="krunolp@gmail.com",
description="MARS: Meta-Learning as Score Matching in the Function Space",
long_description=long_description,
long_description_content_type="text/markdown",
url='github.com/krunolp/mars',
package_dir={'MARS': 'MARS'},
packages=setuptools.find_packages(),
install_requires=[
'numpy',
'pandas',
'scipy',
'sklearn',
'tensorflow>2.8.0',
'tensorflow-probability',
'matplotlib',
'jax',
'jaxlib',
'GPJax',
],
)