Import symmetry utilities and constants in inference_sampler

Author: Wenzhao-protein

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)