diff --git a/act/tests/test_utils.py b/act/tests/test_utils.py index 03e0e3aa56..08b55beefe 100644 --- a/act/tests/test_utils.py +++ b/act/tests/test_utils.py @@ -1,5 +1,6 @@ """ Unit tests for ACT utils module. """ +import importlib import tempfile from datetime import datetime from pathlib import Path @@ -15,10 +16,10 @@ import act -try: - import pyart +spec = importlib.util.find_spec('pyart') +if spec is not None: PYART_AVAILABLE = True -except ImportError: +else: PYART_AVAILABLE = False diff --git a/act/utils/data_utils.py b/act/utils/data_utils.py index a0d879822b..058ecba7a1 100644 --- a/act/utils/data_utils.py +++ b/act/utils/data_utils.py @@ -3,6 +3,7 @@ """ +import importlib import warnings import metpy @@ -11,10 +12,10 @@ import scipy.stats as stats import xarray as xr -try: - import pyart +spec = importlib.util.find_spec('pyart') +if spec is not None: PYART_AVAILABLE = True -except ImportError: +else: PYART_AVAILABLE = False @@ -585,6 +586,8 @@ def create_pyart_obj( raise ImportError( 'Py-ART needs to be installed on your system to convert to ' 'Py-ART Object.' ) + else: + import pyart # Get list of variables if none provided if variables is None: variables = list(obj.keys())