change branch behavior

tcpekin · Feb 3, 2025 · 4e7d000 · 4e7d000
1 parent f260337
commit 4e7d000
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py4DSTEM/process/diffraction/crystal_viz.py b/py4DSTEM/process/diffraction/crystal_viz.py
@@ -150,7 +150,7 @@ def plot_structure(
 
     # atoms
     ID_all = np.unique(ID)
-    if occ is None:
+    if np.all(np.asarray(occ) == 1):
         for ID_plot in ID_all:
             sub = ID == ID_plot
             ax.scatter(