address breaking change

chemosim-lab · Jun 26, 2024 · de8784d · de8784d
1 parent 8bc9ac1
commit de8784d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/prolif/plotting/complex3d.py b/prolif/plotting/complex3d.py
@@ -310,13 +310,15 @@ def compare(
 
     def _populate_view(  # noqa: PLR0912
         self,
-        v: py3Dmol.view,
+        v: Union[py3Dmol.view, Complex3D],
         position: Tuple[int, int] = (0, 0),
         display_all: bool = False,
         colormap: Optional[Dict[ResidueId, str]] = None,
         only_interacting: bool = True,
         remove_hydrogens: Union[bool, Literal["ligand", "protein"]] = True,
     ) -> None:
+        if isinstance(v, Complex3D) and v._view:
+            v = v._view
         self._colormap = {} if colormap is None else colormap
         self._models = {}
         self._mid = -1