forked from rsanchezgarc/deepEMhancer
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
63 lines (56 loc) · 1.73 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
import importlib
import os
import setuptools
from setuptools import setup
def version():
version = importlib.import_module("deepEMhancer").__version__
return version
def readme():
readmePath = os.path.abspath(os.path.join(__file__, "..", "README.md"))
try:
with open(readmePath) as f:
return f.read()
except UnicodeDecodeError:
try:
with open(readmePath, 'r', encoding='utf-8') as f:
return f.read()
except Exception as e:
return "Description not available due to unexpected error: "+str(e)
installTfGpu = os.environ.get("DEEPEMHANCER_INSTALL_GPU", None)
if installTfGpu:
tfTarget='-gpu==1.14.*'
else:
tfTarget='==1.14.*'
install_requires = [
'numpy==1.16.*',
'scikit-image==0.15.*',
'scipy==1.3.1',
'joblib==0.13.*',
'tensorflow%s'%tfTarget,
'keras==2.2.*',
'pandas==0.25.*',
'mrcfile==1.1.2',
'requests==2.22.*',
'tqdm==4.42',
'mrcfile==1.1.2',
'keras-radam== 0.12',
'keras-contrib @ git+https://github.com/keras-team/keras-contrib.git@3fc5ef709e061416f4bc8a92ca3750c824b5d2b0'
]
setup(name='deepEMhancer',
version=version(),
description='Deep learning for cryo-EM maps post-processing',
long_description=readme(),
long_description_content_type="text/markdown",
keywords='cryo-EM deep learning',
url='https://github.com/rsanchezgarc/deepEMhancer',
author='Ruben Sanchez-Garcia',
author_email='[email protected]',
license='Apache 2.0',
packages=setuptools.find_packages(),
install_requires=install_requires,
dependency_links=[],
entry_points={
'console_scripts': ['deepemhancer=deepEMhancer.exeDeepEMhancer:commanLineFun'],
},
include_package_data=True,
zip_safe=False)