forked from ACEsuit/mace
-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.cfg
38 lines (35 loc) · 1.01 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
[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
opt_einsum
ase
torch-ema
prettytable
matscipy
# 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
[options.extras_require]
wandb = wandb