Update utils

Author: JewelRoam

graph_net/torch/utils.py

@@ -132,7 +132,6 @@ def format_data(data):
 
     def process_tensor_info(tensor_info, name_prefix="example_input"):
         data_list = None
-        # MODIFICATION: Handle sparse tensor serialization
         is_sparse = tensor_info.get("type") == "sparse_int_tensor"
         sparse_indices = None
         sparse_values = None
@@ -226,7 +225,6 @@ def convert_meta_classes_to_tensors(file_path):
         data_value = None
         data_type = getattr(torch, attrs.get("dtype", "torch.float").split(".")[-1])
 
-        # MODIFICATION: Reconstruct sparse tensors during loading
         if attrs.get("is_sparse"):
             indices_shape = (len(attrs.get("shape")), -1)
             indices = torch.tensor(attrs["indices"]).reshape(indices_shape)
@@ -274,4 +272,4 @@ def replay_tensor(info):
     std = info["info"]["std"]
     if "data" in info and info["data"] is not None:
         return info["data"].to(device)
-    return torch.randn(size=shape).to(dtype).to(device) * std * 0.2 + mean
+    return torch.randn(size=shape).to(dtype).to(device) * std * 0.2 + mean