Use postponed annotations (PEP 563)

Author: shuds13

optimas/utils/ax/ax_model_manager.py

@@ -1,6 +1,17 @@
 """Contains the definition of the AxModelManager class."""
 
-from typing import Optional, Union, List, Tuple, Dict, Any, Literal
+from __future__ import annotations
+
+from typing import (
+    Optional,
+    Union,
+    List,
+    Tuple,
+    Dict,
+    Any,
+    Literal,
+    TYPE_CHECKING,
+)
 
 import numpy as np
 from numpy.typing import NDArray
@@ -34,6 +45,10 @@
 from optimas.core import VaryingParameter, Objective
 from optimas.utils.other import convert_to_dataframe
 
+if TYPE_CHECKING:
+    from ax.service.ax_client import AxClient
+    from ax.modelbridge.torch import TorchModelBridge
+
 
 class AxModelManager:
     """Class for building and exploring GP models using an ``AxClient``.