feat: #36 Use axis instead of transpose

kausalflow · Jun 24, 2024 · 163eb26 · 163eb26
1 parent 3b36b5b
commit 163eb26
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hamilflow/models/d1/harmonic_oscillator_chain.py b/hamilflow/models/d1/harmonic_oscillator_chain.py
@@ -81,7 +81,7 @@ def _x(self, t: ArrayLike) -> tuple[ArrayLike, ArrayLike]:
  t = np.asarray([t])
  travelling_waves = np.asarray([tw._x(t) for tw in self.travelling_waves])
  # FIXME this is imaginary
- original_xs = ifft(travelling_waves.T, norm="ortho").T
+ original_xs = ifft(travelling_waves, axis=0, norm="ortho")
 
  return original_xs, travelling_waves