-
Notifications
You must be signed in to change notification settings - Fork 9
/
setup.py
24 lines (21 loc) · 857 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
from setuptools import setup, find_packages
import sys
import os
setup(
name='atmorep',
version='0.1',
url='https://github.com/clessig/atmorep',
author='Christian Lessig',
author_email='christian@atmorep.org',
description='AtmoRep',
packages=find_packages(),
# if packages are available in a native form fo the host system then these should be used
install_requires=['torch>=2.3', 'numpy', 'matplotlib', 'zarr', 'pandas', 'typing_extensions', 'pathlib', 'wandb', 'cloudpickle', 'ecmwflibs', 'cfgrib', 'netcdf4', 'xarray', 'pytz', 'torchinfo', 'pytest', 'cfgrib', 'dask'],
data_files=[('./output', []), ('./logs', []), ('./results',[])],
)
if not os.path.exists('./output'):
os.mkdir('./output')
if not os.path.exists('./logs'):
os.mkdir('./logs')
if not os.path.exists('./results'):
os.mkdir('./results')