From 85d4d41cece2f5c7eabf36847b826ce734ac0049 Mon Sep 17 00:00:00 2001 From: Alyssa Travitz Date: Tue, 3 Feb 2026 16:14:13 -0800 Subject: [PATCH] add requires_package for pygraphviz --- src/openfe/tests/utils/test_atommapping_network_plotting.py | 2 -- src/openfe/utils/atommapping_network_plotting.py | 3 ++- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/openfe/tests/utils/test_atommapping_network_plotting.py b/src/openfe/tests/utils/test_atommapping_network_plotting.py index 9e8086f88..d823b8d5d 100644 --- a/src/openfe/tests/utils/test_atommapping_network_plotting.py +++ b/src/openfe/tests/utils/test_atommapping_network_plotting.py @@ -1,9 +1,7 @@ -import importlib.resources import inspect from unittest import mock import matplotlib -import matplotlib.figure import pytest from matplotlib import pyplot as plt diff --git a/src/openfe/utils/atommapping_network_plotting.py b/src/openfe/utils/atommapping_network_plotting.py index 13d899d66..98255c69b 100644 --- a/src/openfe/utils/atommapping_network_plotting.py +++ b/src/openfe/utils/atommapping_network_plotting.py @@ -10,7 +10,7 @@ from rdkit import Chem from openfe import LigandNetwork, SmallMoleculeComponent -from openfe.utils.custom_typing import MPL_MouseEvent +from openfe.utils import requires_package from openfe.utils.network_plotting import Edge, GraphDrawing, Node @@ -156,6 +156,7 @@ class AtomMappingNetworkDrawing(GraphDrawing): EdgeCls = AtomMappingEdge +@requires_package("pygraphviz") def plot_atommapping_network(network: LigandNetwork): """Convenience method for plotting the atom mapping network