diff --git a/rdagent/components/coder/model_coder/benchmark/gt_code/gpsconv.py b/rdagent/components/coder/model_coder/benchmark/gt_code/gpsconv.py index 0fe66a8d..e1b832fe 100644 --- a/rdagent/components/coder/model_coder/benchmark/gt_code/gpsconv.py +++ b/rdagent/components/coder/model_coder/benchmark/gt_code/gpsconv.py @@ -188,7 +188,7 @@ model_cls = GPSConv 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 = GPSConv(channels=node_features.size(-1), conv=MessagePassing())