Fix.

pulquero · Jul 2, 2024 · e5038f5 · e5038f5
1 parent ab5f4e4
commit e5038f5
Show file tree

Hide file tree

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

diff --git a/sgw_tools/graph.py b/sgw_tools/graph.py
@@ -436,7 +436,7 @@ def _eigL(self, spectrum_only):
  # more accurate to decompose the adjacency matrix itself then shift the result
  e, U = util.eigh(self.W, spectrum_only=spectrum_only)
  e = np.flip(1 - e/np.max(e))
- U = np.flip(U, axis=1) if U else None
+ U = np.flip(U, axis=1) if U is not None else None
  return e, U
  else:
  return super()._eigL(spectrum_only=spectrum_only)

diff --git a/tests/test_biggraph.py b/tests/test_biggraph.py
@@ -261,6 +261,10 @@ def test_fourier_basis(self):
  assert len(G.e) == G.N
  assert G.U.shape[1] == G.N
  assert G.e[-1] < 2
+
+ def test_adjacency_fourier_basis(self):
+ G = BigGraph.create_from(graphs.Logo(), lap_type='adjacency')
+ G.compute_fourier_basis()
 
  def test_eigendecompositions(self):
  G = BigGraph.create_from(graphs.Logo())