From 9d34ae7d3b94f69e5a09f587ff7a04eb8bf04be3 Mon Sep 17 00:00:00 2001 From: TPTBusiness Date: Wed, 29 Apr 2026 17:19:26 +0200 Subject: [PATCH] fix(security): add nosec for torch.load (B614 #739) --- rdagent/components/coder/model_coder/gt_code.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rdagent/components/coder/model_coder/gt_code.py b/rdagent/components/coder/model_coder/gt_code.py index dbb2c2f9..49f0e955 100644 --- a/rdagent/components/coder/model_coder/gt_code.py +++ b/rdagent/components/coder/model_coder/gt_code.py @@ -125,7 +125,7 @@ class AntiSymmetricConv(torch.nn.Module): 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