Rename edge_tensors() to get_edge_tensors().

pulquero · Jul 22, 2024 · ca1419b · ca1419b
1 parent 858d299
commit ca1419b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
  name="sgw_tools",
- version="3.0",
+ version="3.0.1",
  author="Mark Hale",
  license="MIT",
  description="Spectral graph wavelet tools",

diff --git a/sgw_torch/__init__.py b/sgw_torch/__init__.py
@@ -9,7 +9,7 @@
 
 from . import layers
 
-def edge_tensors(G):
+def get_edge_tensors(G):
  edge_list = G.get_edge_list()
  edge_index = np.array([
  np.concatenate((edge_list[0], edge_list[1])),

diff --git a/tests/test_torch.py b/tests/test_torch.py
@@ -11,7 +11,7 @@
 class TestCase(unittest.TestCase):
  def test_get_laplacian(self):
  G = sgw.BigGraph.create_from(gsp.graphs.Sensor(50, seed=32), lap_type="adjacency")
- edge_index, edge_weight = sgw_torch.edge_tensors(G)
+ edge_index, edge_weight = sgw_torch.get_edge_tensors(G)
  L_index, L_weight = sgw_torch.get_laplacian(edge_index, edge_weight, lap_type="adjacency", num_nodes=G.N)
  torch_L = torch_utils.to_scipy_sparse_matrix(L_index, L_weight)
  np.testing.assert_allclose(G.L.toarray(), torch_L.toarray())
@@ -30,7 +30,7 @@ def test_ChebLayer(self):
  layer = sgw_torch.layers.ChebLayer(1, 1, K+1, lap_type="normalized", bias=False)
  for c, p in zip(coeffs, layer.parameters()):
  p.data = torch.tensor([[c]])
- edge_index, edge_weight = sgw_torch.edge_tensors(G)
+ edge_index, edge_weight = sgw_torch.get_edge_tensors(G)
  y = layer(torch.from_numpy(s), edge_index, edge_weight)
 
  np.testing.assert_allclose(expected, y.detach().numpy().squeeze())