Spaces:
Sleeping
Sleeping
File size: 2,018 Bytes
2efd5b6 |
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 |
# -*- coding: utf-8 -*-
import os
import re
from setuptools import setup
def read(filename):
with open(os.path.join(os.path.dirname(__file__), filename)) as f:
file_content = f.read()
return file_content
def get_requirements():
requirements = []
for requirement in read('requirements.txt').splitlines():
if requirement.startswith('git+') or requirement.startswith('svn+') or requirement.startswith('hg+'):
parsed_requires = re.findall(r'#egg=([\w\d\.]+)-([\d\.]+)$', requirement)
if parsed_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)
return requirements
def get_links():
return [
requirement for requirement in read('requirements.txt').splitlines()
if requirement.startswith('git+') or requirement.startswith('svn+') or requirement.startswith('hg+')
]
def get_version():
""" Get version from the package without actually importing it. """
init = read('deepfloyd_if/__init__.py')
for line in init.split('\n'):
if line.startswith('__version__'):
return eval(line.split('=')[1])
setup(
name='deepfloyd_if',
version=get_version(),
author='DeepFloyd, StabilityAI',
author_email='[email protected]',
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',
)
|