diff --git a/rdagent/components/coder/model_coder/benchmark/gt_code/linkx.py b/rdagent/components/coder/model_coder/benchmark/gt_code/linkx.py
index 2ba31a35..332c2075 100644
--- a/rdagent/components/coder/model_coder/benchmark/gt_code/linkx.py
+++ b/rdagent/components/coder/model_coder/benchmark/gt_code/linkx.py
@@ -170,7 +170,7 @@ class LINKX(torch.nn.Module):
 model_cls = LINKX
 
 if __name__ == "__main__":
-    node_features = torch.load("node_features.pt", weights_only=True)
+    node_features = torch.load("node_features.pt", weights_only=True)  # nosec B614
     edge_index = torch.load("edge_index.pt", weights_only=True)  # nosec B614
 
     # Model instantiation and forward pass