map2map/setup.py

23 lines
491 B
Python

from setuptools import setup, find_packages
setup(
name='map2map',
version='0.0',
description='Neural network emulators to transform field data',
author='Yin Li et al.',
author_email='eelregit@gmail.com',
packages=find_packages(),
scripts=[
'scripts/m2m.py',
],
python_requires='>=3.6',
install_requires=[
'torch>=1.2',
'numpy',
'scipy',
],
extras_require={
"visualization": ["tensorboard"],
},
)