use get_cell

pyiron · May 20, 2024 · e21bb2f · e21bb2f
1 parent 99a28ac
commit e21bb2f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/structuretoolkit/visualize.py b/structuretoolkit/visualize.py
@@ -9,6 +9,8 @@
 from typing import Optional
 from scipy.interpolate import interp1d
 
+from structuretoolkit.common.helper import get_cell
+
 __author__ = "Joerg Neugebauer, Sudarsan Surendralal"
 __copyright__ = (
     "Copyright 2021, Max-Planck-Institut für Eisenforschung GmbH - "
@@ -166,7 +168,7 @@ def _get_box_skeleton(cell: np.ndarray):
 
 
 def _draw_box_plotly(fig, structure, px, go):
-    cell = structure.cell
+    cell = get_cell(structure)
     data = fig.data
     for lines in _get_box_skeleton(cell):
         fig = px.line_3d(**{xx: vv for xx, vv in zip(["x", "y", "z"], lines.T)})