-
Notifications
You must be signed in to change notification settings - Fork 251
/
setup.py
35 lines (32 loc) · 722 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
33
34
from setuptools import setup, find_packages
# Minimum dependencies required prior to installation
INSTALL_REQUIRES = [
"charset-normalizer",
"matplotlib",
"openai",
'torch<=2.0.0',
'numpy==1.20.0',
'ray>=1.1.0',
'tensorboard>=1.14.0',
'tensorboardX>=1.6',
'setproctitle',
'psutil',
'pyyaml',
"gym==0.23.1",
"omegaconf",
"termcolor",
"hydra-core>=1.1",
"pyvirtualdisplay",
]
# Installation operation
setup(
name="eureka",
author="Jason Ma",
version="1.0",
description="Eureka",
keywords=["llm", "rl"],
include_package_data=True,
python_requires=">=3.7",
install_requires=INSTALL_REQUIRES,
packages=find_packages("."),
)