From 3cbb62aeb8a66b421edddc06bcd825df8c40b8bc Mon Sep 17 00:00:00 2001 From: Wenzhao Dai Date: Fri, 5 Dec 2025 23:57:33 +0100 Subject: [PATCH 1/2] Import symmetry utilities and constants in inference_sampler --- models/rfd3/src/rfd3/model/inference_sampler.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/models/rfd3/src/rfd3/model/inference_sampler.py b/models/rfd3/src/rfd3/model/inference_sampler.py index 9ea24069..b5bcb30f 100644 --- a/models/rfd3/src/rfd3/model/inference_sampler.py +++ b/models/rfd3/src/rfd3/model/inference_sampler.py @@ -12,6 +12,12 @@ uniform_random_rotation, ) +from rfd3.inference.symmetry.atom_array import ( + FIXED_ENTITY_ID, + FIXED_TRANSFORM_ID +) +from rfd3.inference.symmetry.symmetry_utils import apply_symmetry_to_xyz_atomwise + ranked_logger = RankedLogger(__name__, rank_zero_only=True) From 0ad75294a7cac3e6314df71a72561fd7e6cb04d9 Mon Sep 17 00:00:00 2001 From: Wenzhao Dai Date: Sat, 6 Dec 2025 00:18:11 +0100 Subject: [PATCH 2/2] Simplify import statements in inference_sampler.py --- models/rfd3/src/rfd3/model/inference_sampler.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/models/rfd3/src/rfd3/model/inference_sampler.py b/models/rfd3/src/rfd3/model/inference_sampler.py index 66092a00..1e17dc77 100644 --- a/models/rfd3/src/rfd3/model/inference_sampler.py +++ b/models/rfd3/src/rfd3/model/inference_sampler.py @@ -13,10 +13,7 @@ uniform_random_rotation, ) -from rfd3.inference.symmetry.atom_array import ( - FIXED_ENTITY_ID, - FIXED_TRANSFORM_ID -) +from rfd3.inference.symmetry.atom_array import FIXED_ENTITY_ID, FIXED_TRANSFORM_ID from rfd3.inference.symmetry.symmetry_utils import apply_symmetry_to_xyz_atomwise ranked_logger = RankedLogger(__name__, rank_zero_only=True)