-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
121 lines (104 loc) · 4.14 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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
import os, sys
from setuptools import setup, find_packages
from setuptools.command.install import install
here = os.path.abspath(os.path.dirname(__file__))
# Get the long description from the README file
with open(os.path.join(here, 'README.md'), encoding='utf-8') as f:
long_description = f.read()
normal_install_require_list = [
'numpy', 'torch', 'torchvision',
'tqdm', 'scipy', 'Pillow', 'scikit-learn',
'natsort', 'matplotlib', 'opencv-python',
'bokeh', 'piexif', 'pandas', 'packaging',
'pytorch-lightning', 'pydot', 'argparse',
'pydbus', 'lightglue', 'fastdist', 'flask',
'requests',
]
normal_install_versioned_require_list = [
'numpy==1.21.0', 'torch==2.0.1', 'torchvision==0.15.2',
'tqdm==4.64.1', 'scipy==1.3.3', 'Pillow==10.2.0', 'scikit-learn==0.22.2.post1',
'natsort==8.3.1', 'matplotlib==3.7.3', 'opencv-python==4.7.0.72',
'bokeh==3.0.3', 'piexif==1.1.3', 'pandas==1.5.3',
'packaging==24.0', 'pytorch-lightning',#'pytorch-lightning==1.5.10', # throws an error :/
'pydbus==0.6.0', 'argparse==1.4.0', 'pydot==1.4.1',
'lightglue==0.0', 'fastdist==1.1.5', 'flask==2.3.2',
'requests==2.22.0',
]
# ROS Packages:
ros_install_require_list = [
'cv-bridge', 'rospkg', 'rospy',
'rospy_message_converter', 'aarapsi_robot_pack',
'genpy', 'rosbag', 'tf',
]
ros_install_versioned_require_list = [
'cv-bridge==1.16.2', 'rospkg==1.5.0', 'rospy==1.16.0',
'rospy_message_converter==0.5.9', 'aarapsi_robot_pack',
'genpy==0.6.15', 'rosbag==1.16.0', 'tf==1.13.2',
]
# # workaround as opencv-python does not show up in "pip list" within a conda environment
# # we do not care as conda recipe has py-opencv requirement anyhow
# is_conda = os.path.exists(os.path.join(sys.prefix, 'conda-meta'))
# if not is_conda:
# install_require_list.append('opencv-python')
# EXTRAS_MISSING = False
# class CustomInstallCommand(install):
# def finalize_options(self):
# super().finalize_options()
# if not any(arg.startswith('[') for arg in sys.argv):
# global EXTRAS_MISSING
# EXTRAS_MISSING = True
setup(
name='pyaarapsi',
version='0.2.2',
description='PyAARAPSI: Python Package for AARAPSI Project',
author='Owen Claxton, Connor Malone, and Helen Carson',
author_email='[email protected]',
url='https://github.com/QVPR/pyaarapsi',
license='MIT',
classifiers=[
# 3 - Alpha
# 4 - Beta
# 5 - Production/Stable
'Development Status :: 3 - Alpha',
# Indicate who your project is intended for
'Intended Audience :: Developers',
# Pick your license as you wish (should match "license" above)
'License :: OSI Approved :: MIT License',
# Specify the Python versions you support here. In particular, ensure
# that you indicate whether you support Python 2, Python 3 or both.
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
'Programming Language :: Python :: 3.9',
],
python_requires='>=3.6',
extras_require={
'basic': normal_install_require_list,
'basic-versioned': normal_install_versioned_require_list,
'ros': ros_install_require_list,
'ros-versioned': ros_install_versioned_require_list,
'complete': normal_install_require_list + ros_install_require_list,
'complete-versioned': normal_install_versioned_require_list + ros_install_versioned_require_list
},
packages=find_packages(),
keywords=[
'python', 'place recognition', 'image retrieval', 'computer vision', 'robotics'
],
# cmdclass={
# 'install': CustomInstallCommand,
# }
)
# _message = \
# """
# No dependency mode selected!
# Please try again with one of the following options:
# >>> pip install -e .[basic]
# >>> pip install -e .[basic,ros]
# >>> pip install -e .[basic-versioned]
# >>> pip install -e .[basic-versioned,ros-versioned]
# >>> pip install -e .[complete]
# >>> pip install -e .[complete-versioned]
# """
# if EXTRAS_MISSING:
# # raise Exception(_message)
# pass