forked from ACEsuit/mace
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.cfg
56 lines (53 loc) · 1.34 KB
/
setup.cfg
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
[metadata]
name = mace-torch
version = attr: mace.__version__
short_description = MACE - Fast and accurate machine learning interatomic potentials with higher order equivariant message passing.
long_description = file: README.md
long_description_content_type = text/markdown
url = https://github.com/ACEsuit/mace
classifiers =
Programming Language :: Python :: 3
Operating System :: OS Independent
License :: OSI Approved :: MIT License
[options]
packages = find:
python_requires = >=3.7
install_requires =
torch>=1.12
e3nn==0.4.4
numpy<2.0
opt_einsum
ase
torch-ema
prettytable
matscipy
h5py
torchmetrics
python-hostlist
configargparse
GitPython
pyYAML
tqdm
# for plotting:
matplotlib
pandas
[options.entry_points]
console_scripts =
mace_active_learning_md = mace.cli.active_learning_md:main
mace_create_lammps_model = mace.cli.create_lammps_model:main
mace_eval_configs = mace.cli.eval_configs:main
mace_plot_train = mace.cli.plot_train:main
mace_run_train = mace.cli.run_train:main
mace_prepare_data = mace.cli.preprocess_data:main
mace_finetuning = mace.cli.fine_tuning_select:main
[options.extras_require]
wandb = wandb
fpsample = fpsample
dev =
black
isort
mypy
pre-commit
pytest
pylint
schedulefree = schedulefree