Skip to content

Commit fae79bd

Browse files
authored
Add SPICE 1.1.4, remove leftover remove_ref_energy in some datasets (#303)
* Add SPICE 1.1.4 Remove bogus remove_ref_energy param * Remove a couple of remove_ref_energy instances in Datasets
1 parent 2429c94 commit fae79bd

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

torchmdnet/datasets/genentech.py

-1
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,6 @@ def __init__(
4343
transform,
4444
pre_transform,
4545
pre_filter,
46-
remove_ref_energy=False,
4746
properties=("y"),
4847
)
4948

torchmdnet/datasets/qm9q.py

-1
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,6 @@ def __init__(
4646
transform,
4747
pre_transform,
4848
pre_filter,
49-
remove_ref_energy=False,
5049
properties=("y", "neg_dy", "q", "pq", "dp"),
5150
)
5251

torchmdnet/datasets/spice.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313

1414

1515
class SPICE(MemmappedDataset):
16-
1716
"""
1817
SPICE dataset (https://github.com/openmm/spice-dataset)
1918
@@ -65,6 +64,10 @@ class SPICE(MemmappedDataset):
6564
"url": "https://zenodo.org/record/7606550/files",
6665
"file": "SPICE-1.1.3.hdf5",
6766
},
67+
"1.1.4": {
68+
"url": "https://zenodo.org/records/8222043/files",
69+
"file": "SPICE-1.1.4.hdf5",
70+
},
6871
}
6972

7073
@property
@@ -103,7 +106,6 @@ def __init__(
103106
transform,
104107
pre_transform,
105108
pre_filter,
106-
remove_ref_energy=False,
107109
properties=("y", "neg_dy"),
108110
)
109111

0 commit comments

Comments
 (0)