We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1bbb03a commit eede5d7Copy full SHA for eede5d7
examples/TensorNet-ANI2X.yaml
@@ -39,6 +39,7 @@ num_rbf: 32
39
num_workers: 4
40
output_model: Scalar
41
precision: 32
42
+prior_model: null
43
remove_ref_energy: true
44
rbf_type: expnorm
45
redirect: false
tests/test_examples.py
@@ -19,7 +19,7 @@ def test_example_yamls(fname):
19
args = yaml.load(f, Loader=yaml.FullLoader)
20
21
prior = None
22
- if "prior_model" in args and args["prior_model"] is not None:
+ if args["prior_model"] is not None:
23
dataset = DummyDataset(has_atomref=True)
24
prior = getattr(priors, args["prior_model"])(dataset=dataset)
25
0 commit comments