Skip to content

Commit eede5d7

Browse files
committed
revert removal of prior_model: null in ANI-2x config file as it causes too many test failures
1 parent 1bbb03a commit eede5d7

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

examples/TensorNet-ANI2X.yaml

+1
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ num_rbf: 32
3939
num_workers: 4
4040
output_model: Scalar
4141
precision: 32
42+
prior_model: null
4243
remove_ref_energy: true
4344
rbf_type: expnorm
4445
redirect: false

tests/test_examples.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def test_example_yamls(fname):
1919
args = yaml.load(f, Loader=yaml.FullLoader)
2020

2121
prior = None
22-
if "prior_model" in args and args["prior_model"] is not None:
22+
if args["prior_model"] is not None:
2323
dataset = DummyDataset(has_atomref=True)
2424
prior = getattr(priors, args["prior_model"])(dataset=dataset)
2525

0 commit comments

Comments
 (0)