forked from py4dstem/py4DSTEM
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
57 lines (54 loc) · 1.88 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
from setuptools import setup, find_packages
from distutils.util import convert_path
with open("README.md","r") as f:
long_description = f.read()
version_ns = {}
vpath = convert_path('py4DSTEM/version.py')
with open(vpath) as version_file:
exec(version_file.read(), version_ns)
setup(
name='py4DSTEM',
version=version_ns['__version__'],
packages=find_packages(),
description='An open source python package for processing and analysis of 4D STEM data.',
long_description=long_description,
long_description_content_type="text/markdown",
url='https://github.com/py4dstem/py4DSTEM/',
author='Benjamin H. Savitzky',
author_email='[email protected]',
license='GNU GPLv3',
keywords="STEM 4DSTEM",
python_requires='>=3.8',
install_requires=[
'numpy >= 1.19',
'scipy >= 1.5.2',
'h5py >= 3.2.0',
'ncempy >= 1.8.1',
'matplotlib >= 3.2.2',
'scikit-image >= 0.17.2',
'scikit-learn >= 0.23.2',
'ipywidgets >= 7.6.3',
'tqdm >= 4.46.1',
'dill >= 0.3.3',
'gdown >= 4.4.0',
'dask >= 2.3.0',
'distributed >= 2.3.0',
'emdfile >= 0.0.4',
],
extras_require={
'ipyparallel': ['ipyparallel >= 6.2.4', 'dill >= 0.3.3'],
'cuda': ['cupy'],
'acom': ['pymatgen >= 2022', 'mp-api == 0.24.1'],
'aiml': ['tensorflow == 2.4.1','tensorflow-addons <= 0.14.0','crystal4D'],
'aiml-cuda': ['tensorflow == 2.4.1','tensorflow-addons <= 0.14.0','crystal4D','cupy'],
'numba': ['numba >= 0.49.1']
},
entry_points={
'console_scripts': ['py4DSTEM=py4DSTEM.gui.runGUI:launch']
},
package_data={
'py4DSTEM':['process/utils/scattering_factors.txt',
'process/diskdetection/multicorr_row_kernel.cu',
'process/diskdetection/multicorr_col_kernel.cu']
},
)