Update solve.py

loganbvh · Sep 18, 2023 · cc08591 · cc08591
1 parent 86d326e
commit cc08591
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tdgl/solver/solve.py b/tdgl/solver/solve.py
@@ -225,7 +225,7 @@ def current_func(t):
             areas = cupy.asarray(areas)
             edge_centers = cupy.asarray(edge_centers)
             sites = cupy.asarray(sites)
-            new_A_induced = cupy.empty((num_edges, 2), dtype=float)
+            new_A_induced = cupy.empty((len(edges), 2), dtype=float)
 
     # Running list of the max abs change in |psi|^2 between subsequent solve steps.
     # This list is used to calculate the adaptive time step.