pyg-nightly 2.7.0.dev20250607__py3-none-any.whl → 2.7.0.dev20250608__py3-none-any.whl

torch_geometric/utils/_spmm.py

@@ -63,18 +63,20 @@ def spmm(
 
         # Always convert COO to CSR for more efficient processing:
         if src.layout == torch.sparse_coo:
-            warnings.warn(f"Converting sparse tensor to CSR format for more "
-                          f"efficient processing. Consider converting your "
-                          f"sparse tensor to CSR format beforehand to avoid "
-                          f"repeated conversion (got '{src.layout}')")
+            warnings.warn(
+                f"Converting sparse tensor to CSR format for more "
+                f"efficient processing. Consider converting your "
+                f"sparse tensor to CSR format beforehand to avoid "
+                f"repeated conversion (got '{src.layout}')", stacklevel=2)
             src = src.to_sparse_csr()
 
         # Warn in case of CSC format without gradient computation:
         if src.layout == torch.sparse_csc and not other.requires_grad:
-            warnings.warn(f"Converting sparse tensor to CSR format for more "
-                          f"efficient processing. Consider converting your "
-                          f"sparse tensor to CSR format beforehand to avoid "
-                          f"repeated conversion (got '{src.layout}')")
+            warnings.warn(
+                f"Converting sparse tensor to CSR format for more "
+                f"efficient processing. Consider converting your "
+                f"sparse tensor to CSR format beforehand to avoid "
+                f"repeated conversion (got '{src.layout}')", stacklevel=2)
 
         # Use the default code path for `sum` reduction (works on CPU/GPU):
         if reduce == 'sum':
@@ -99,10 +101,11 @@ def spmm(
         # TODO The `torch.sparse.mm` code path with the `reduce` argument does
         # not yet support CSC :(
         if src.layout == torch.sparse_csc:
-            warnings.warn(f"Converting sparse tensor to CSR format for more "
-                          f"efficient processing. Consider converting your "
-                          f"sparse tensor to CSR format beforehand to avoid "
-                          f"repeated conversion (got '{src.layout}')")
+            warnings.warn(
+                f"Converting sparse tensor to CSR format for more "
+                f"efficient processing. Consider converting your "
+                f"sparse tensor to CSR format beforehand to avoid "
+                f"repeated conversion (got '{src.layout}')", stacklevel=2)
             src = src.to_sparse_csr()
 
         return torch.sparse.mm(src, other, reduce)

torch_geometric/utils/convert.py

@@ -251,13 +251,13 @@ def from_networkx(
     if group_edge_attrs is not None and not isinstance(group_edge_attrs, list):
         group_edge_attrs = edge_attrs
 
-    for i, (_, feat_dict) in enumerate(G.nodes(data=True)):
+    for _, feat_dict in G.nodes(data=True):
         if set(feat_dict.keys()) != set(node_attrs):
             raise ValueError('Not all nodes contain the same attributes')
         for key, value in feat_dict.items():
             data_dict[str(key)].append(value)
 
-    for i, (_, _, feat_dict) in enumerate(G.edges(data=True)):
+    for _, _, feat_dict in G.edges(data=True):
         if set(feat_dict.keys()) != set(edge_attrs):
             raise ValueError('Not all edges contain the same attributes')
         for key, value in feat_dict.items():

torch_geometric/utils/embedding.py

@@ -42,7 +42,8 @@ def get_embeddings(
             hook_handles.append(module.register_forward_hook(hook))
 
     if len(hook_handles) == 0:
-        warnings.warn("The 'model' does not have any 'MessagePassing' layers")
+        warnings.warn("The 'model' does not have any 'MessagePassing' layers",
+                      stacklevel=2)
 
     training = model.training
     model.eval()
@@ -123,8 +124,9 @@ def get_embeddings_hetero(
                 hook_handles.append(module.register_forward_hook(hook))
 
     if len(hook_handles) == 0:
-        warnings.warn("The 'model' does not have any heterogenous "
-                      "'MessagePassing' layers")
+        warnings.warn(
+            "The 'model' does not have any heterogenous "
+            "'MessagePassing' layers", stacklevel=2)
 
     # Run the model forward pass
     training = model.training

torch_geometric/utils/geodesic.py

@@ -66,9 +66,10 @@ def geodesic_distance(  # noqa: D417
 
     if 'dest' in kwargs:
         dst = kwargs['dest']
-        warnings.warn("'dest' attribute in 'geodesic_distance' is deprecated "
-                      "and will be removed in a future release. Use the 'dst' "
-                      "argument instead.")
+        warnings.warn(
+            "'dest' attribute in 'geodesic_distance' is deprecated "
+            "and will be removed in a future release. Use the 'dst' "
+            "argument instead.", stacklevel=2)
 
     max_distance = float('inf') if max_distance is None else max_distance
 

torch_geometric/utils/sparse.py

@@ -70,8 +70,9 @@ def dense_to_sparse(
                          f"three-dimensional (got {adj.dim()} dimensions)")
 
     if mask is not None and adj.dim() == 2:
-        warnings.warn("Mask should not be provided in case the dense "
-                      "adjacency matrix is two-dimensional")
+        warnings.warn(
+            "Mask should not be provided in case the dense "
+            "adjacency matrix is two-dimensional", stacklevel=2)
         mask = None
 
     if mask is not None and mask.dim() != 2: