More helpful error message on errors with snowflake

MolSSI · Dec 13, 2024 · 369c9ed · 369c9ed
1 parent 16a5048
commit 369c9ed
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/qcfractal/qcfractal/snowflake.py b/qcfractal/qcfractal/snowflake.py
@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import importlib
 import logging
 import logging.handlers
 import multiprocessing
@@ -22,12 +23,12 @@
 from .port_util import find_open_port
 from .postgres_harness import create_snowflake_postgres
 
-try:
-    from qcfractalcompute.compute_manager import ComputeManager
-    from qcfractalcompute.config import FractalComputeConfig, FractalServerSettings, LocalExecutorConfig
-except ImportError:
+if importlib.util.find_spec("qcfractalcompute") is None:
     raise RuntimeError("qcfractalcompute is not installed. Snowflake is useless without it")
 
+from qcfractalcompute.compute_manager import ComputeManager
+from qcfractalcompute.config import FractalComputeConfig, FractalServerSettings, LocalExecutorConfig
+
 if TYPE_CHECKING:
     from typing import Dict, Any, Sequence, Optional, Set