From 0f4f956584ce0f777edecf72773699b3ac95251a Mon Sep 17 00:00:00 2001
From: Zach Sherman <zsherman@anl.gov>
Date: Mon, 5 Dec 2022 16:37:01 -0600
Subject: [PATCH] FIX: Fixes call of pyart to supress warning. (#588)

---
 act/tests/test_utils.py | 7 ++++---
 act/utils/data_utils.py | 9 ++++++---
 2 files changed, 10 insertions(+), 6 deletions(-)

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())