FIX: Fixes call of pyart to supress warning. (#588)

ARM-DOE · Dec 5, 2022 · 0f4f956 · 0f4f956
1 parent 47e47bc
commit 0f4f956
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git 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
@@ -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())