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 3e7927db..2ba31a35 100644 --- a/rdagent/components/coder/model_coder/benchmark/gt_code/linkx.py +++ b/rdagent/components/coder/model_coder/benchmark/gt_code/linkx.py @@ -171,7 +171,7 @@ model_cls = LINKX 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 = LINKX(