diff --git a/rdagent/components/coder/model_coder/gt_code.py b/rdagent/components/coder/model_coder/gt_code.py
index 87ad2e59..dbb2c2f9 100644
--- a/rdagent/components/coder/model_coder/gt_code.py
+++ b/rdagent/components/coder/model_coder/gt_code.py
@@ -126,7 +126,7 @@ class AntiSymmetricConv(torch.nn.Module):
 
 if __name__ == "__main__":
     node_features = torch.load("node_features.pt", weights_only=True)
-    edge_index = torch.load("edge_index.pt", weights_only=True)
+    edge_index = torch.load("edge_index.pt", weights_only=True)  # nosec B614
 
     # Model instantiation and forward pass
     model = AntiSymmetricConv(in_channels=node_features.size(-1))