IF/setup.py at develop · hellomierkat/IF · GitHub

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
# -*- coding: utf-8 -*-
importos
importre
fromsetuptoolsimportsetup


defread(filename):
withopen(os.path.join(os.path.dirname(__file__), filename)) asf:
file_content=f.read()
returnfile_content


defget_requirements():
requirements= []
forrequirementinread('requirements.txt').splitlines():
ifrequirement.startswith('git+') orrequirement.startswith('svn+') orrequirement.startswith('hg+'):
parsed_requires=re.findall(r'#egg=([\w\d\.]+)-([\d\.]+)$', requirement)
ifparsed_requires:
package, version=parsed_requires[0]
requirements.append(f'{package}=={version}')
else:
print('WARNING! For correct matching dependency links need to specify package name and version'
'such as <dependency url>#egg=<package_name>-<version>')
else:
requirements.append(requirement)
returnrequirements


defget_links():
return [
requirementforrequirementinread('requirements.txt').splitlines()
ifrequirement.startswith('git+') orrequirement.startswith('svn+') orrequirement.startswith('hg+')
 ]


defget_version():
""" Get version from the package without actually importing it. """
init=read('deepfloyd_if/__init__.py')
forlineininit.split('\n'):
ifline.startswith('__version__'):
returneval(line.split('=')[1])


setup(
name='deepfloyd_if',
version=get_version(),
author='DeepFloyd, StabilityAI',
author_email='shonenkov@gmail.com',
description='DeepFloyd-IF (Imagen Free)',
packages=['deepfloyd_if', 'deepfloyd_if/model', 'deepfloyd_if/modules', 'deepfloyd_if/pipelines',
'deepfloyd_if/resources'],
package_data={'deepfloyd_if/resources': ['*.png', '*.npy', '*.npz']},
install_requires=get_requirements(),
dependency_links=get_links(),
long_description=read('README.md'),
long_description_content_type='text/markdown',
)