forked from allegroai/clearml
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
106 lines (94 loc) · 3.94 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
"""
ClearML - Artificial Intelligence Version Control
https://github.com/allegroai/clearml
"""
import os.path
# Always prefer setuptools over distutils
from setuptools import setup, find_packages
import codecs
def read_text(filepath):
with codecs.open(filepath, "r", encoding="utf-8") as f:
return f.read()
here = os.path.dirname(__file__)
# Get the long description from the README file
long_description = read_text(os.path.join(here, 'README.md'))
# fix github, dark logo hack.
long_description = long_description.replace(
"""<img align="center" src="docs/clearml-logo.svg#gh-light-mode-only" alt="Clear|ML"><img align="center" src="docs/clearml-logo-dark.svg#gh-dark-mode-only" alt="Clear|ML">""", # noqa
"""<a href="https://app.clear.ml"><img src="https://github.com/allegroai/clearml/blob/master/docs/clearml-logo.svg?raw=true" width="250px"></a>""", # noqa
1
)
def read_version_string(version_file):
for line in read_text(version_file).splitlines():
if line.startswith('__version__'):
delim = '"' if '"' in line else "'"
return line.split(delim)[1]
else:
raise RuntimeError("Unable to find version string.")
version = read_version_string("clearml/version.py")
requirements = read_text(os.path.join(here, 'requirements.txt')).splitlines()
setup(
name='clearml',
version=version,
description='ClearML - Auto-Magical Experiment Manager, Version Control, and MLOps for AI',
long_description=long_description,
long_description_content_type='text/markdown',
# The project's main homepage.
url='https://github.com/allegroai/clearml',
author='ClearML',
author_email='[email protected]',
license='Apache License 2.0',
classifiers=[
'Development Status :: 5 - Production/Stable',
'Intended Audience :: Developers',
'Intended Audience :: Science/Research',
'Operating System :: POSIX :: Linux',
'Operating System :: MacOS :: MacOS X',
'Operating System :: Microsoft',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
'Topic :: Software Development',
'Topic :: Software Development :: Version Control',
'Topic :: System :: Logging',
'Topic :: System :: Monitoring',
'Programming Language :: Python :: 2.7',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
'Programming Language :: Python :: 3.9',
'Programming Language :: Python :: 3.10',
'Programming Language :: Python :: 3.11',
'Programming Language :: Python :: 3.12',
'License :: OSI Approved :: Apache Software License',
],
keywords='clearml trains development machine deep learning version control machine-learning machinelearning '
'deeplearning deep-learning experiment-manager experimentmanager',
packages=find_packages(exclude=['contrib', 'docs', 'data', 'examples', 'tests']),
install_requires=requirements,
extras_require={
's3': [
'boto3>=1.9',
],
'azure': [
'azure-storage-blob>=12.0.0',
],
'gs': [
'google-cloud-storage>=1.13.2',
],
},
package_data={
'clearml': ['config/default/*.conf', 'backend_api/config/default/*.conf']
},
include_package_data=True,
# To provide executable scripts, use entry points in preference to the
# "scripts" keyword. Entry points provide cross-platform support and allow
# pip to create the appropriate form of executable for the target platform.
entry_points={
'console_scripts': [
'clearml-init = clearml.cli.config.__main__:main',
'clearml-data = clearml.cli.data.__main__:main',
'clearml-task = clearml.cli.task.__main__:main',
'clearml-param-search = clearml.cli.hpo.__main__:main'
],
},
)