Restore from __future__ import annotations

- Required for forward type inference

Author: tomvothecoder

e3sm_diags/driver/utils/dataset_xr.py

@@ -8,6 +8,8 @@
 variables from dataset files.
 """
 
+from __future__ import annotations
+
 import collections
 import fnmatch
 import glob

e3sm_diags/driver/utils/regrid.py

@@ -1,4 +1,6 @@
-from typing import Literal
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Literal
 
 import dask
 import numpy as np
@@ -8,7 +10,9 @@
 from e3sm_diags.derivations.default_regions_xr import REGION_SPECS
 from e3sm_diags.driver import FRAC_REGION_KEYS
 from e3sm_diags.logger import _setup_child_logger
-from e3sm_diags.parameter.core_parameter import CoreParameter
+
+if TYPE_CHECKING:
+    from e3sm_diags.parameter.core_parameter import CoreParameter
 
 logger = _setup_child_logger(__name__)
 

e3sm_diags/parameter/core_parameter.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import copy
 import importlib
 import sys

e3sm_diags/plot/lat_lon_plot.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import TYPE_CHECKING
 
 import cartopy.crs as ccrs

tests/e3sm_diags/driver/test_tc_analysis_driver.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import shutil
 import tempfile
 from unittest import TestCase