chore: fix plot

Exabyte-io · May 31, 2024 · b13ff7d · b13ff7d
1 parent 2cfaee5
commit b13ff7d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/utils/plot.py b/utils/plot.py
@@ -17,7 +17,8 @@ def plot_strain_vs_atoms(interfaces: List[Material], settings: Dict[str, Union[s
     data = []
     for index, interface in enumerate(interfaces):
         strain_percentage = interface.metadata[StrainModes.mean_abs_strain] * 100
-        num_sites = len(interface.basis.values())
+        print(interface.basis["coordinates"])
+        num_sites = len(interface.basis["coordinates"])
 
         hover_text = f"Index: {index}<br>Strain: {strain_percentage:.2f}%<br>Atoms: {num_sites}"