fix cmap import (#880)

* fix cmap import * Update visualization_3D.py
OpenFreeEnergy · Jul 3, 2024 · 6aaf7e3 · 6aaf7e3
1 parent 43eccaa
commit 6aaf7e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openfe/utils/visualization_3D.py b/openfe/utils/visualization_3D.py
@@ -84,7 +84,7 @@ def _add_spheres(view:py3Dmol.view, mol1:Chem.Mol, mol2:Chem.Mol, mapping:Dict[i
         mapping of atoms from mol1 to mol2
     """
     # Get colourmap of size mapping
-    cmap = plt.cm.get_cmap("hsv", len(mapping))
+    cmap = plt.get_cmap("hsv", len(mapping))
     for i, pair in enumerate(mapping.items()):
         p1 = mol1.GetConformer().GetAtomPosition(pair[0])
         p2 = mol2.GetConformer().GetAtomPosition(pair[1])