Skip to content

Commit 22c34aa

Browse files
authored
Merge branch 'RosettaCommons:production' into production
2 parents ca14a87 + 8da6dc7 commit 22c34aa

2 files changed

Lines changed: 3 additions & 0 deletions

File tree

models/rfd3/src/rfd3/engine.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -391,6 +391,8 @@ def _multiply_specifications(
391391
design_specifications = {}
392392
for prefix, example_spec in inputs.items():
393393
# Record task name in the specification
394+
if "extra" not in example_spec:
395+
example_spec["extra"] = {}
394396
example_spec["extra"]["task_name"] = prefix
395397

396398
# ... Create n_batches for example

models/rfd3/src/rfd3/model/inference_sampler.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
rot_vec_mul,
1313
uniform_random_rotation,
1414
)
15+
from rfd3.inference.symmetry.symmetry_utils import apply_symmetry_to_xyz_atomwise
1516

1617
ranked_logger = RankedLogger(__name__, rank_zero_only=True)
1718

0 commit comments

Comments
 (0)