From ee29ed2368630bdf30f9aeb10a79ff68252e4f4e Mon Sep 17 00:00:00 2001 From: TPTBusiness Date: Wed, 29 Apr 2026 17:27:53 +0200 Subject: [PATCH] fix(security): add nosec for torch.load (B614 #735) --- rdagent/components/coder/model_coder/benchmark/gt_code/pmlp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rdagent/components/coder/model_coder/benchmark/gt_code/pmlp.py b/rdagent/components/coder/model_coder/benchmark/gt_code/pmlp.py index 7582577c..de93fda5 100644 --- a/rdagent/components/coder/model_coder/benchmark/gt_code/pmlp.py +++ b/rdagent/components/coder/model_coder/benchmark/gt_code/pmlp.py @@ -102,7 +102,7 @@ class PMLP(torch.nn.Module): model_cls = PMLP 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