From a6d1cb78fe22363b2524ad09802a6433c9216930 Mon Sep 17 00:00:00 2001
From: "Matthew W. Thompson" <mattwthompson@protonmail.com>
Date: Thu, 13 Jul 2023 09:17:58 -0500
Subject: [PATCH] Add `requires_openeye` decorators

---
 .pre-commit-config.yaml                           | 2 +-
 openff/bespokefit/executor/utilities/depiction.py | 3 +++
 openff/bespokefit/utilities/molecule.py           | 3 +++
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index a904853b..c55f50cf 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -7,7 +7,7 @@ repos:
   - id: check-yaml
   - id: debug-statements
 - repo: https://github.com/psf/black
-  rev: 23.3.0
+  rev: 23.7.0
   hooks:
   - id: black
     files: ^openff
diff --git a/openff/bespokefit/executor/utilities/depiction.py b/openff/bespokefit/executor/utilities/depiction.py
index fe6a3fd8..c0a8422a 100644
--- a/openff/bespokefit/executor/utilities/depiction.py
+++ b/openff/bespokefit/executor/utilities/depiction.py
@@ -1,11 +1,14 @@
 import functools
 from typing import Optional, Tuple
 
+from openff.utilities.utilities import requires_oe_module
+
 IMAGE_UNAVAILABLE_SVG = """
 <svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" style="isolation:isolate" viewBox="0 0 200 200" width="200pt" height="200pt"><defs><clipPath id="_clipPath_eSdCSpw1sB1xWp7flmMoZ0WjTPwPpzQh"><rect width="200" height="200"/></clipPath></defs><g clip-path="url(#_clipPath_eSdCSpw1sB1xWp7flmMoZ0WjTPwPpzQh)"><g clip-path="url(#_clipPath_LvpdWbrYj1cREqoXz8Lwbk3ZilfC6tg9)"><text transform="matrix(1,0,0,1,44.039,91.211)" style="font-family:'Open Sans';font-weight:400;font-size:30px;font-style:normal;fill:#000000;stroke:none;">Preview</text><text transform="matrix(1,0,0,1,17.342,132.065)" style="font-family:'Open Sans';font-weight:400;font-size:30px;font-style:normal;fill:#000000;stroke:none;">Unavailable</text></g><defs><clipPath id="_clipPath_LvpdWbrYj1cREqoXz8Lwbk3ZilfC6tg9"><rect x="0" y="0" width="166" height="81.709" transform="matrix(1,0,0,1,17,59.146)"/></clipPath></defs></g></svg>
 """
 
 
+@requires_oe_module("oechem")
 def _oe_smiles_to_image(smiles: str, highlight_atoms: Tuple[int]) -> str:
     from openeye import oechem, oedepict
 
diff --git a/openff/bespokefit/utilities/molecule.py b/openff/bespokefit/utilities/molecule.py
index fd0f16bf..42f84d5d 100644
--- a/openff/bespokefit/utilities/molecule.py
+++ b/openff/bespokefit/utilities/molecule.py
@@ -3,8 +3,10 @@
 
 from openff.toolkit.topology import Molecule
 from openff.toolkit.utils.exceptions import ToolkitUnavailableException
+from openff.utilities.utilities import requires_oe_module
 
 
+@requires_oe_module("oechem")
 def _oe_get_atom_symmetries(molecule: Molecule) -> List[int]:
     from openeye import oechem
 
@@ -31,6 +33,7 @@ def get_atom_symmetries(molecule: Molecule) -> List[int]:
         return _rd_get_atom_symmetries(molecule)
 
 
+@requires_oe_module("oechem")
 def _oe_canonical_atom_order(molecule: Molecule) -> List[int]:
     from openeye import oechem