This repository has been archived by the owner on Sep 18, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 67
/
setup.py
72 lines (68 loc) · 2.53 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
from setuptools import find_packages, setup
# package name
_name = 'finetuner'
# package long description
try:
with open('README.md', encoding='utf8') as fp:
_long_description = fp.read()
except FileNotFoundError:
_long_description = ''
if __name__ == '__main__':
setup(
name=_name,
packages=find_packages(exclude=['*.tests', '*.tests.*', 'tests.*', 'tests']),
include_package_data=True,
description='Task-oriented finetuning for better embeddings on neural search.',
author='Jina AI',
author_email='hello@jina.ai',
url='https://github.com/jina-ai/finetuner/',
license='Apache 2.0',
download_url='https://github.com/jina-ai/finetuner/tags',
long_description=_long_description,
long_description_content_type='text/markdown',
zip_safe=False,
setup_requires=['setuptools>=18.0', 'wheel'],
install_requires=[
'docarray[common]<0.30.0',
'finetuner-stubs==0.13.10',
'finetuner-commons==0.13.10',
],
extras_require={
'full': [
'jina-hubble-sdk==0.33.1',
'trimesh==3.16.4',
],
'test': [
'black==23.3.0',
'flake8==6.0.0',
'isort==5.12.0',
'pytest==7.0.0',
'pytest-cov==3.0.0',
'pytest-mock==3.7.0',
],
},
python_requires='>=3.8.0',
classifiers=[
'Development Status :: 5 - Production/Stable',
'Intended Audience :: Developers',
'Intended Audience :: Education',
'Intended Audience :: Science/Research',
'Programming Language :: Python :: 3.8',
'Programming Language :: Python :: 3.9',
'Programming Language :: Python :: 3.10',
'License :: OSI Approved :: Apache Software License',
'Environment :: Console',
'Operating System :: OS Independent',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
],
project_urls={
'Documentation': 'https://finetuner.jina.ai',
'Source': 'https://github.com/jina-ai/finetuner/',
'Tracker': 'https://github.com/jina-ai/finetuner/issues',
},
keywords=(
'jina neural-search neural-network deep-learning pretraining '
'fine-tuning pretrained-models triplet-loss metric-learning '
'siamese-network few-shot-learning'
),
)