We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents ca14a87 + 8da6dc7 commit 22c34aaCopy full SHA for 22c34aa
2 files changed
models/rfd3/src/rfd3/engine.py
@@ -391,6 +391,8 @@ def _multiply_specifications(
391
design_specifications = {}
392
for prefix, example_spec in inputs.items():
393
# Record task name in the specification
394
+ if "extra" not in example_spec:
395
+ example_spec["extra"] = {}
396
example_spec["extra"]["task_name"] = prefix
397
398
# ... Create n_batches for example
models/rfd3/src/rfd3/model/inference_sampler.py
@@ -12,6 +12,7 @@
12
rot_vec_mul,
13
uniform_random_rotation,
14
)
15
+from rfd3.inference.symmetry.symmetry_utils import apply_symmetry_to_xyz_atomwise
16
17
ranked_logger = RankedLogger(__name__, rank_zero_only=True)
18
0 commit comments