Update utils

JewelRoam · JewelRoam · commit 70936431bc45 · 2025-08-08T09:21:11.000Z
diff --git a/graph_net/torch/utils.py b/graph_net/torch/utils.py
@@ -98,6 +98,7 @@ def handle_named_tensors(tensor):
         key: handle_named_tensors(tensor) for key, tensor in state_dict.items()
     }
 
+    # dynamic_shapes = extract_dynamic_shapes(example_inputs)
     return {
         "input_info": processed_inputs,
         "weight_info": processed_weights,

Original file line number	Diff line number	Diff line change
`@@ -98,6 +98,7 @@ def handle_named_tensors(tensor):`
`98`	`98`	`key: handle_named_tensors(tensor) for key, tensor in state_dict.items()`
`99`	`99`	`}`
`100`	`100`
	`101`	`+ # dynamic_shapes = extract_dynamic_shapes(example_inputs)`
`101`	`102`	`return {`
`102`	`103`	`"input_info": processed_inputs,`
`103`	`104`	`"weight_info": processed_weights,`