fix: find singleton indices

pyamg · Oct 12, 2023 · e00bf8e · e00bf8e
1 parent 4b9502f
commit e00bf8e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyamg/vis/vis_coarse.py b/pyamg/vis/vis_coarse.py
@@ -125,7 +125,7 @@ def vis_aggregate_groups(V, E2V, AggOp, mesh_type,
     # 3.5 #
     # single node aggregates
     sums = np.array(AggOp.sum(axis=0)).ravel()
-    E2V_c = np.where(sums == 1)[0]
+    E2V_c = np.argwhere(np.isin(AggOp.indices, np.where(sums == 1)[0])).ravel()
     Nel_c = len(E2V_c)
 
     # 4 #