File size: 4,221 Bytes
89cd41b
 
 
 
 
f1eddde
76e941d
89cd41b
 
 
f1eddde
 
c4ff088
89cd41b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
e473a42
3eda6d3
49d0cfc
 
05e8129
221dfe3
3eda6d3
639bd5d
89cd41b
 
221dfe3
cf512f3
b22d49d
639bd5d
 
f856a5e
1d1ee87
cf512f3
3eda6d3
d937c80
3eda6d3
 
0d1ce35
3eda6d3
cf512f3
 
b22d49d
b9692e2
cf512f3
1c7cd6c
cf512f3
52c1bfb
cf512f3
8396dce
f1eddde
cf512f3
b9692e2
699356f
b22d49d
cf512f3
1c7cd6c
 
 
b22d49d
 
 
221dfe3
89cd41b
 
e473a42
 
 
0b5acc7
9d1a2a5
0b5acc7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
e473a42
0b5acc7
 
 
 
 
 
221dfe3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1effaf5
aa496c2
221dfe3
0b5acc7
 
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
[build-system]
requires=["flit_core >=3.2,<4"]
build-backend="flit_core.buildapi"

[project]
name="mlip-arena"
version="0.0.1a1"
authors=[
    {name="Yuan Chiang", email="cyrusyc@lbl.gov"},
]
description="Fair and transparent benchmark of machine-learned interatomic potentials (MLIPs), beyond basic error metrics"
readme=".github/README.md"
requires-python=">=3.10"
keywords=[
    "pytorch",
    "machine-learning-interatomic-potentials",
    "huggingface",
    "deep-learning",
    "graph-neural-networks",
]
classifiers=[
    "Development Status :: 1 - Planning",
    "Programming Language :: Python",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Programming Language :: Python :: 3 :: Only",
]
dependencies=[
    "ase",
    "torch",
    "torch_dftd>=0.4.0",
    "huggingface_hub",
    "datasets",
    "torch-geometric",
    "safetensors",
    "pymatgen",
]

[project.optional-dependencies]
run = [
    "prefect>=3.0.4,<3.1.0",
    "prefect-dask",
    "dask",
    "dask_jobqueue",
    "tables",
]
app = [
    "streamlit==1.38.0",
    "plotly",
    "bokeh==2.4.3",
    "statsmodels"
]
test = [
    "torch==2.2.0",
    "nvidia-ml-py==12.560.30",
    "e3nn==0.5.0",
    "matgl==1.1.2",
    "dgl==2.4.0",
    "chgnet==0.3.8",
    "fairchem-core==1.2.0",
    "sevenn==0.9.3.post1",
    "orb-models==0.4.0",
    "pynanoflann@git+https://github.com/dwastberg/pynanoflann#egg=af434039ae14bedcbb838a7808924d6689274168",
    "alignn==2024.5.27",
    "mattersim==1.0.0rc9",
    "pytest",
    "prefect>=3.0.4,<3.1.0",
]
mace = [
    "mace-torch==0.3.4",
]
deepmd = [
    "deepmd-kit@git+https://github.com/deepmodeling/deepmd-kit.git@v3.0.0b4"
]

[project.urls]
Homepage = "https://github.com/atomind-ai/mlip-arena"
Issues = "https://github.com/atomind-ai/mlip-arena/issues"

[tool.ruff]
# Exclude a variety of commonly ignored directories.
extend-include = ["*.ipynb"]
exclude = [
    ".bzr",
    ".direnv",
    ".eggs",
    ".git",
    ".git-rewrite",
    ".hg",
    ".ipynb_checkpoints",
    ".mypy_cache",
    ".nox",
    ".pants.d",
    ".pyenv",
    ".pytest_cache",
    ".pytype",
    ".ruff_cache",
    ".svn",
    ".tox",
    ".venv",
    ".vscode",
    "__pypackages__",
    "_build",
    "buck-out",
    "build",
    "dist",
    "node_modules",
    "site-packages",
    "venv",
]

# Same as Black.
line-length = 88
indent-width = 4

[tool.ruff.lint]
select = [
    "B",      # flake8-bugbear
    "C4",     # flake8-comprehensions
    "E",      # pycodestyle error
    "EXE",    # flake8-executable
    "F",      # pyflakes
    "FA",     # flake8-future-annotations
    "FBT003", # boolean-positional-value-in-call
    "FLY",    # flynt
    "I",      # isort
    "ICN",    # flake8-import-conventions
    "PD",     # pandas-vet
    "PERF",   # perflint
    "PIE",    # flake8-pie
    "PL",     # pylint
    "PT",     # flake8-pytest-style
    "PYI",    # flakes8-pyi
    "Q",      # flake8-quotes
    "RET",    # flake8-return
    "RSE",    # flake8-raise
    "RUF",    # Ruff-specific rules
    "SIM",    # flake8-simplify
    "SLOT",   # flake8-slots
    "TCH",    # flake8-type-checking
    "TID",    # tidy imports
    "TID",    # flake8-tidy-imports
    "UP",     # pyupgrade
    "W",      # pycodestyle warning
    "YTT",    # flake8-2020
]
ignore = [
    "C408",   # Unnecessary dict call
    "PLR",    # Design related pylint codes
    "E501",   # Line too long
    "B028",   # No explicit stacklevel
    "EM101",  # Exception must not use a string literal
    "EM102",  # Exception must not use an f-string literal
    "G004",   # f-string in Logging statement
    "RUF015", # Prefer next(iter())
    "RET505", # Unnecessary `elif` after `return`
    "PT004",  # Fixture does not return anthing
    "B017",   # pytest.raises
    "PT011",  # pytest.raises
    "PT012",  # pytest.raises"
    "E741",  # ambigous variable naming, i.e. one letter
    "FBT003",  # boolean positional variable in function call
    "PERF203",  # `try`-`except` within a loop incurs performance overhead (no overhead in Py 3.11+)
    "F405",  # 'module' may be undefined, or defined from star imports
    "C0301",  # Line too long
]
fixable = ["ALL"]
pydocstyle.convention = "google"