forked from vimalabs/VIMABench
-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
59 lines (49 loc) · 1.6 KB
/
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
import pathlib
import pkg_resources
from setuptools import setup, find_packages
PKG_NAME = "vima_bench"
VERSION = "0.0.1"
EXTRAS = {}
def _read_file(fname):
# this_dir = os.path.abspath(os.path.dirname(__file__))
# with open(os.path.join(this_dir, fname)) as f:
with pathlib.Path(fname).open() as fp:
return fp.read()
def _read_install_requires():
with pathlib.Path("requirements.txt").open() as fp:
return [
str(requirement) for requirement in pkg_resources.parse_requirements(fp)
]
def _fill_extras(extras):
if extras:
extras["all"] = list(set([item for group in extras.values() for item in group]))
return extras
setup(
name=PKG_NAME,
version=VERSION,
author="VIMA-Bench Developers",
# url='http://github.com/',
description="research project",
long_description=_read_file("README.md"),
long_description_content_type="text/markdown",
keywords=["Deep Learning", "Machine Learning"],
license="Apache License, Version 2.0",
packages=find_packages(include=f"{PKG_NAME}.*")
+ find_packages(include="vimasim.*"),
include_package_data=True,
zip_safe=False,
entry_points={
"console_scripts": [
# 'cmd_tool=mylib.subpkg.module:main',
]
},
install_requires=_read_install_requires(),
extras_require=_fill_extras(EXTRAS),
python_requires=">=3.7",
classifiers=[
"Development Status :: 3 - Alpha",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Environment :: Console",
"Programming Language :: Python :: 3",
],
)