From 7f8f9f0ead81bf98d0e1152c858a7aa521bebb0e Mon Sep 17 00:00:00 2001
From: Lorenzo Tarricone <96067929+LorenzoTarricone@users.noreply.github.com>
Date: Sat, 6 Dec 2025 16:26:41 +0000
Subject: [PATCH] Import apply_symmetry_to_xyz_atomwise function

apply_symmetry_to_xyz_atomwise function used in line 361 (new 362) when generating symmetric protein, but not imported
---
 models/rfd3/src/rfd3/model/inference_sampler.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/models/rfd3/src/rfd3/model/inference_sampler.py b/models/rfd3/src/rfd3/model/inference_sampler.py
index 2c1482f..e11b288 100644
--- a/models/rfd3/src/rfd3/model/inference_sampler.py
+++ b/models/rfd3/src/rfd3/model/inference_sampler.py
@@ -12,6 +12,7 @@ from foundry.utils.rotation_augmentation import (
     rot_vec_mul,
     uniform_random_rotation,
 )
+from rfd3.inference.symmetry.symmetry_utils import apply_symmetry_to_xyz_atomwise
 
 ranked_logger = RankedLogger(__name__, rank_zero_only=True)