classiq 0.52.0__py3-none-any.whl → 0.54.0__py3-none-any.whl

classiq/__init__.py

@@ -1,7 +1,5 @@
 """Classiq SDK."""
 
-from typing import List
-
 from classiq.interface._version import VERSION as _VERSION
 from classiq.interface.generator.application_apis import *  # noqa: F403
 from classiq.interface.generator.arith.register_user_input import (
@@ -117,5 +115,5 @@ __all__ = (
 )
 
 
-def __dir__() -> List[str]:
+def __dir__() -> list[str]:
     return __all__

classiq/_analyzer_extras/_ipywidgets_async_extension.py

@@ -1,6 +1,7 @@
 import asyncio
+from collections.abc import Awaitable
 from functools import wraps
-from typing import TYPE_CHECKING, Awaitable, Callable
+from typing import TYPE_CHECKING, Callable
 
 from ipywidgets import widgets  # type: ignore[import]
 

classiq/_internals/__init__.py

@@ -1,20 +0,0 @@
-import sys
-import warnings
-
-from classiq.interface.exceptions import ClassiqDeprecationWarning
-
-
-def _check_python_version() -> None:
-    if sys.version_info >= (3, 9):
-        return
-    warnings.warn(  # type: ignore[unreachable]
-        "Python version 3.8 is expected to reach its end-of-life on October 2024.\n"
-        "See https://devguide.python.org/versions/#supported-versions\n"
-        "The Classiq SDK is expected to drop support for 3.8 around the same time.\n"
-        "Please upgrade to a newer version of Python to avoid issues in the future.",
-        ClassiqDeprecationWarning,
-        stacklevel=2,
-    )
-
-
-_check_python_version()

classiq/_internals/api_wrapper.py

@@ -1,5 +1,5 @@
 import json
-from typing import Any, Dict, List, Optional, Protocol, Type, TypeVar
+from typing import Any, Optional, Protocol, TypeVar
 
 import pydantic
 
@@ -52,7 +52,7 @@ class StatusType(Protocol):
 
 def _parse_job_response(
     job_result: JobDescription[JSONObject],
-    output_type: Type[ResultType],
+    output_type: type[ResultType],
 ) -> ResultType:
     if job_result.result is not None:
         return output_type.model_validate(job_result.result)
@@ -84,10 +84,10 @@ class ApiWrapper:
         cls,
         http_method: str,
         url: str,
-        body: Optional[Dict] = None,
-        params: Optional[Dict] = None,
+        body: Optional[dict] = None,
+        params: Optional[dict] = None,
         use_versioned_url: bool = True,
-        headers: Optional[Dict[str, str]] = None,
+        headers: Optional[dict[str, str]] = None,
         allow_none: bool = False,
     ) -> dict:
         res: Any = await client().call_api(
@@ -295,10 +295,10 @@ class ApiWrapper:
 
     @staticmethod
     def _get_devices_from_hardware_info(
-        hardware_info: List[HardwareInformation],
+        hardware_info: list[HardwareInformation],
         params: analysis_params.AnalysisOptionalDevicesParams,
     ) -> analysis_result.DevicesResult:
-        available_hardware: Dict[Provider, Dict[str, bool]] = {
+        available_hardware: dict[Provider, dict[str, bool]] = {
             Provider.IBM_QUANTUM: {},
             Provider.AMAZON_BRAKET: {},
             Provider.AZURE_QUANTUM: {},
@@ -318,7 +318,7 @@ class ApiWrapper:
         )
 
     @classmethod
-    async def call_get_all_hardware_devices(cls) -> List[HardwareInformation]:
+    async def call_get_all_hardware_devices(cls) -> list[HardwareInformation]:
         data = await client().call_api(
             http_method=HTTPMethod.GET,
             url="/hardware-catalog/v1/hardwares",
@@ -342,19 +342,25 @@ class ApiWrapper:
     @classmethod
     async def call_iqcc_init_auth(cls, data: IQCCInitAuthData) -> IQCCInitAuthResponse:
         response = await cls._call_task_pydantic(
-            http_method=HTTPMethod.GET,
+            http_method=HTTPMethod.PUT,
             url=f"{routes.IQCC_INIT_AUTH_FULL_PATH}",
             model=data,
         )
-        return IQCCInitAuthResponse.parse_obj(response)
+        return IQCCInitAuthResponse.model_validate(response)
 
     @classmethod
     async def call_iqcc_probe_auth(
         cls, data: IQCCProbeAuthData
-    ) -> IQCCProbeAuthResponse:
-        response = await cls._call_task_pydantic(
-            http_method=HTTPMethod.GET,
-            url=f"{routes.IQCC_PROBE_AUTH_FULL_PATH}",
-            model=data,
-        )
-        return IQCCProbeAuthResponse.parse_obj(response)
+    ) -> Optional[IQCCProbeAuthResponse]:
+        try:
+            response = await cls._call_task_pydantic(
+                http_method=HTTPMethod.PUT,
+                url=f"{routes.IQCC_PROBE_AUTH_FULL_PATH}",
+                model=data,
+            )
+        except ClassiqAPIError as ex:
+            if ex.status_code == 418:
+                return None
+            raise
+
+        return IQCCProbeAuthResponse.model_validate(response)

classiq/_internals/async_utils.py

@@ -3,12 +3,10 @@ import functools
 import itertools
 import logging
 import time
+from collections.abc import AsyncGenerator, Awaitable, Iterable
 from typing import (
     Any,
-    AsyncGenerator,
-    Awaitable,
     Callable,
-    Iterable,
     Optional,
     SupportsFloat,
     TypeVar,

classiq/_internals/authentication/auth0.py

@@ -1,6 +1,6 @@
 import urllib.parse
 from dataclasses import dataclass
-from typing import Any, Dict, Optional, Union
+from typing import Any, Optional, Union
 
 from httpx import AsyncClient, Response, codes
 from pydantic import Field
@@ -56,9 +56,9 @@ class Auth0:
     async def _make_request(
         self,
         url: str,
-        payload: Dict[str, str],
+        payload: dict[str, str],
         allow_error: Union[bool, int] = False,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         encoded_payload = urllib.parse.urlencode(payload)
         client: AsyncClient
         async with AsyncClient(
@@ -76,7 +76,7 @@ class Auth0:
             f"Request to Auth0 failed with error code {code}: {data.get('error')}"
         )
 
-    async def get_device_data(self, get_refresh_token: bool = True) -> Dict[str, Any]:
+    async def get_device_data(self, get_refresh_token: bool = True) -> dict[str, Any]:
         payload = {
             "client_id": self._auth_settings.client_id,
             "audience": self._auth_settings.audience,
@@ -89,7 +89,7 @@ class Auth0:
             payload=payload,
         )
 
-    async def poll_tokens(self, device_code: str) -> Dict[str, Any]:
+    async def poll_tokens(self, device_code: str) -> dict[str, Any]:
         payload = {
             "client_id": self._client_id,
             "device_code": device_code,

classiq/_internals/authentication/device.py

@@ -1,7 +1,8 @@
 import asyncio
 import webbrowser
+from collections.abc import Iterable
 from datetime import timedelta
-from typing import Any, Dict, Iterable, Optional, TypeVar
+from typing import Any, Optional, TypeVar
 
 from classiq.interface.exceptions import (
     ClassiqAuthenticationError,
@@ -26,7 +27,7 @@ class DeviceRegistrar:
         cls, get_refresh_token: bool = True, text_only: bool = False
     ) -> Tokens:
         auth0_client = Auth0()
-        data: Dict[str, Any] = await auth0_client.get_device_data(
+        data: dict[str, Any] = await auth0_client.get_device_data(
             get_refresh_token=get_refresh_token
         )
 
@@ -48,7 +49,7 @@ class DeviceRegistrar:
 
     @classmethod
     def _handle_ready_data(
-        cls, data: Dict[str, Any], get_refresh_token: bool
+        cls, data: dict[str, Any], get_refresh_token: bool
     ) -> Tokens:
         access_token: Optional[str] = data.get("access_token")
         # If refresh token was not requested, this would be None
@@ -72,7 +73,7 @@ class DeviceRegistrar:
         timeout: float,
         get_refresh_token: bool = True,
     ) -> Tokens:
-        async def poller() -> Dict[str, Any]:
+        async def poller() -> dict[str, Any]:
             nonlocal device_code
             return await auth0_client.poll_tokens(device_code=device_code)
 

classiq/_internals/authentication/password_manager.py

@@ -5,7 +5,7 @@ import os
 import pathlib
 import platform
 import stat
-from typing import Any, Dict, Optional
+from typing import Any, Optional
 
 import keyring
 from keyring.backends import fail
@@ -124,12 +124,12 @@ class FilePasswordManager(PasswordManager):
         super().__init__()
         self.credentials_file = pathlib.Path(self._CLASSIQ_CREDENTIALS_FILE_PATH)
 
-    def _update_file(self, token_dict: Dict) -> None:
+    def _update_file(self, token_dict: dict) -> None:
         self.credentials_file.touch()
         self.credentials_file.chmod(stat.S_IRUSR | stat.S_IWUSR)
         self.credentials_file.write_text(json.dumps(token_dict))
 
-    def _get_token_dict(self) -> Dict:
+    def _get_token_dict(self) -> dict:
         if self.credentials_file.exists():
             return json.loads(self.credentials_file.read_text())
         return {}

classiq/_internals/authentication/token_manager.py

@@ -2,7 +2,8 @@ import argparse
 import logging
 import threading
 import warnings
-from typing import Optional, Sequence, Type
+from collections.abc import Sequence
+from typing import Optional
 
 from classiq.interface.exceptions import (
     ClassiqAuthenticationError,
@@ -14,7 +15,7 @@ from classiq._internals.authentication.auth0 import Auth0
 from classiq._internals.authentication.device import DeviceRegistrar, Tokens
 from classiq._internals.config import Configuration
 
-PASSWORD_MANAGERS: Sequence[Type[pm.PasswordManager]] = [
+PASSWORD_MANAGERS: Sequence[type[pm.PasswordManager]] = [
     pm.KeyringPasswordManager,
     pm.FilePasswordManager,
 ]

classiq/_internals/client.py

@@ -6,12 +6,10 @@ import os
 import platform
 import ssl
 import sys
+from collections.abc import Awaitable
 from typing import (
     Any,
-    Awaitable,
     Callable,
-    Dict,
-    List,
     NoReturn,
     Optional,
     TypeVar,
@@ -37,7 +35,7 @@ _logger = logging.getLogger(__name__)
 
 _RETRY_COUNT = 2
 
-Headers = Dict[str, str]
+Headers = dict[str, str]
 
 APPROVED_API_ERROR_MESSAGES_FOR_RESTART = [
     "Call to API failed with code 502",
@@ -203,9 +201,9 @@ class Client:
             message += f": {detail}"
         except Exception:  # noqa: S110
             pass
-        raise ClassiqAPIError(message)
+        raise ClassiqAPIError(message, response.status_code)
 
-    def _make_client_args(self) -> Dict[str, Any]:
+    def _make_client_args(self) -> dict[str, Any]:
         return {
             "base_url": str(self._config.host),
             "timeout": self._HTTP_TIMEOUT_SECONDS,
@@ -217,11 +215,11 @@ class Client:
         self,
         http_method: str,
         url: str,
-        body: Optional[Dict] = None,
-        params: Optional[Dict] = None,
+        body: Optional[dict] = None,
+        params: Optional[dict] = None,
         use_versioned_url: bool = True,
-        headers: Optional[Dict[str, str]] = None,
-    ) -> Union[Dict, List, str]:
+        headers: Optional[dict[str, str]] = None,
+    ) -> Union[dict, list, str]:
         if use_versioned_url:
             url = self.make_versioned_url(url)
         async with self.async_client() as async_client:
@@ -239,10 +237,10 @@ class Client:
         self,
         http_method: str,
         url: str,
-        body: Optional[Dict] = None,
-        headers: Optional[Dict] = None,
+        body: Optional[dict] = None,
+        headers: Optional[dict] = None,
         use_versioned_url: bool = True,
-    ) -> Union[Dict, str]:
+    ) -> Union[dict, str]:
         if use_versioned_url:
             url = self.make_versioned_url(url)
         with httpx.Client(**self._make_client_args()) as sync_client:

classiq/_internals/config.py

@@ -2,7 +2,7 @@
 
 import os
 import pathlib
-from typing import List, Optional, Union
+from typing import Optional, Union
 
 import configargparse  # type: ignore[import]
 import pydantic
@@ -53,7 +53,7 @@ if os.name == "posix":
     ] + _DEFAULT_CONFIG_FILES
 
 
-def init(args: Optional[Union[str, List[str]]] = None) -> Configuration:
+def init(args: Optional[Union[str, list[str]]] = None) -> Configuration:
     """Initialize the configuration object.
 
     Args:

classiq/_internals/jobs.py

@@ -1,6 +1,7 @@
 import json
 import logging
-from typing import Callable, Dict, Iterable, Optional, Set, TypeVar
+from collections.abc import Iterable
+from typing import Callable, Optional, TypeVar
 
 import httpx
 import pydantic
@@ -52,9 +53,9 @@ class JobPoller:
     def __init__(
         self,
         base_url: str,
-        required_headers: Optional[Set[str]] = None,
+        required_headers: Optional[set[str]] = None,
         use_versioned_url: bool = True,
-        additional_headers: Optional[Dict[str, str]] = None,
+        additional_headers: Optional[dict[str, str]] = None,
     ) -> None:
         self._required_headers = required_headers or set()
         self._additional_headers = additional_headers
@@ -88,7 +89,7 @@ class JobPoller:
 
     @try_again_on_failure
     async def _request(
-        self, http_method: str, url: str, body: Optional[Dict] = None
+        self, http_method: str, url: str, body: Optional[dict] = None
     ) -> httpx.Response:
         # Update headers in case they change
         self._async_client.headers.update(client().get_headers())
@@ -98,7 +99,7 @@ class JobPoller:
         client().handle_response(response)
         return response
 
-    async def _submit(self, body: Dict) -> httpx.Response:
+    async def _submit(self, body: dict) -> httpx.Response:
         return await self._request(http_method="POST", url=self._base_url, body=body)
 
     def _interval_sec(self) -> Iterable[float]:
@@ -152,7 +153,7 @@ class JobPoller:
         await self._request(http_method="PUT", url=cancel_url)
 
     async def run(
-        self, body: Dict, timeout_sec: Optional[float]
+        self, body: dict, timeout_sec: Optional[float]
     ) -> GeneralJobDescription:
         async with self._async_client:
             submit_response = await self._submit(body=body)

classiq/_internals/type_validation.py

@@ -1,4 +1,4 @@
-from typing import Any, Tuple, Type, TypeVar, Union, overload
+from typing import Any, TypeVar, Union, overload
 
 T = TypeVar("T")
 U = TypeVar("U")
@@ -10,37 +10,37 @@ U = TypeVar("U")
 
 @overload
 def validate_type(
-    obj: Any, expected_type: Type[T], operation: str, exception_type: Type[Exception]
+    obj: Any, expected_type: type[T], operation: str, exception_type: type[Exception]
 ) -> T: ...
 
 
 @overload
 def validate_type(
     obj: Any,
-    expected_type: Tuple[Type[T], Type[U]],
+    expected_type: tuple[type[T], type[U]],
     operation: str,
-    exception_type: Type[Exception],
+    exception_type: type[Exception],
 ) -> Union[T, U]: ...
 
 
 @overload
 def validate_type(
     obj: Any,
-    expected_type: Tuple[Type[T], ...],
+    expected_type: tuple[type[T], ...],
     operation: str,
-    exception_type: Type[Exception],
+    exception_type: type[Exception],
 ) -> Any: ...
 
 
 def validate_type(
     obj: Any,
-    expected_type: Union[Tuple[type, ...], type],
+    expected_type: Union[tuple[type, ...], type],
     operation: str,
-    exception_type: Type[Exception],
+    exception_type: type[Exception],
 ) -> Any:
     operation = operation[0].upper() + operation[1:]
     if not isinstance(obj, expected_type):
-        expected_types: Tuple[type, ...]
+        expected_types: tuple[type, ...]
         if isinstance(expected_type, type):
             expected_types = (expected_type,)
         else:

classiq/analyzer/__init__.py

@@ -1,5 +1,3 @@
-from typing import List
-
 import classiq.analyzer.show_interactive_hack
 from classiq.analyzer.analyzer import Analyzer
 
@@ -8,5 +6,5 @@ from ..analyzer import rb
 __all__ = ["rb"]
 
 
-def __dir__() -> List[str]:
+def __dir__() -> list[str]:
     return ["rb"]

classiq/analyzer/analyzer.py

@@ -2,8 +2,9 @@
 
 import json
 import webbrowser
+from collections.abc import Sequence
 from importlib.util import find_spec
-from typing import Any, Dict, List, Optional, Sequence, Union
+from typing import Any, Optional, Union
 from urllib.parse import urljoin
 
 import plotly.graph_objects as go
@@ -79,8 +80,8 @@ class Analyzer(AnalyzerUtilities):
         )
 
     def get_available_devices(
-        self, providers: Optional[List[ProviderNameEnum]] = None
-    ) -> Dict[ProviderNameEnum, List[DeviceName]]:
+        self, providers: Optional[list[ProviderNameEnum]] = None
+    ) -> dict[ProviderNameEnum, list[DeviceName]]:
         """Deprecated. Use get_all_hardware_devices instead.
 
         Returns dict of the available devices by the providers. only devices
@@ -133,7 +134,7 @@ class Analyzer(AnalyzerUtilities):
     def get_hardware_comparison_table(
         self,
         providers: Optional[Sequence[Union[str, AnalyzerProviderVendor]]] = None,
-        devices: Optional[List[str]] = None,
+        devices: Optional[list[str]] = None,
     ) -> None:
         """create a comparison table between the transpiled circuits result on different hardware.
         The  comparison table included the depth, multi qubit gates count,and total gates count of the circuits.
@@ -159,8 +160,8 @@ class Analyzer(AnalyzerUtilities):
 
     def plot_hardware_comparison_table(
         self,
-        providers: Optional[List[Union[str, AnalyzerProviderVendor]]] = None,
-        devices: Optional[List[str]] = None,
+        providers: Optional[list[Union[str, AnalyzerProviderVendor]]] = None,
+        devices: Optional[list[str]] = None,
     ) -> None:
         """plot the comparison table. if it has not been created it, it first creates the table using all the
         available hardware.
@@ -174,7 +175,7 @@ class Analyzer(AnalyzerUtilities):
     def _hardware_comparison_condition(
         self,
         providers: Optional[Sequence[Union[str, AnalyzerProviderVendor]]] = None,
-        devices: Optional[List[str]] = None,
+        devices: Optional[list[str]] = None,
     ) -> None:
         if (
             providers is not None

classiq/analyzer/analyzer_utilities.py

@@ -1,4 +1,4 @@
-from typing import Dict, List, Union
+from typing import Union
 
 from classiq.interface.analyzer import analysis_params
 from classiq.interface.backend.quantum_backend_providers import AnalyzerProviderVendor
@@ -13,9 +13,9 @@ ProviderLowerName = str
 HardwareGraphJson = str
 Availability = bool
 ProviderNameEnum = Union[ProviderName, AnalyzerProviderVendor]
-AvailableDevices = Dict[DeviceName, Availability]
-ProviderAvailableDevices = Dict[ProviderLowerName, AvailableDevices]
-HardwareGraphs = Dict[DeviceName, HardwareGraphJson]
+AvailableDevices = dict[DeviceName, Availability]
+ProviderAvailableDevices = dict[ProviderLowerName, AvailableDevices]
+HardwareGraphs = dict[DeviceName, HardwareGraphJson]
 
 
 class AnalyzerUtilities:
@@ -32,7 +32,7 @@ class AnalyzerUtilities:
         self.hardware_graphs = hardware_graphs
 
     async def _request_available_devices_async(
-        self, providers: List[ProviderNameEnum]
+        self, providers: list[ProviderNameEnum]
     ) -> None:
         requested_providers = self._requested_providers_filter(providers)
         if not requested_providers:
@@ -69,8 +69,8 @@ class AnalyzerUtilities:
             )
 
     def _requested_providers_filter(
-        self, providers: List[ProviderNameEnum]
-    ) -> List[ProviderNameEnum]:
+        self, providers: list[ProviderNameEnum]
+    ) -> list[ProviderNameEnum]:
         return list(
             filter(
                 lambda provider: self.available_devices.get(_to_lower_case(provider))
@@ -81,7 +81,7 @@ class AnalyzerUtilities:
 
     def _filter_devices_by_qubits_count(
         self, provider: ProviderNameEnum
-    ) -> List[DeviceName]:
+    ) -> list[DeviceName]:
         device_avail_dict = self.available_devices[_to_lower_case(provider)]
         return list(
             filter(lambda device: device_avail_dict[device], device_avail_dict.keys())

classiq/analyzer/rb.py

@@ -1,5 +1,5 @@
 from enum import Enum as PythonEnum
-from typing import Dict, List, Optional, Set, Union
+from typing import Optional, Union
 
 import matplotlib.pyplot as plt
 import numpy as np
@@ -15,7 +15,7 @@ from classiq.executor import BackendPreferencesAndResult
 
 
 class RBAnalysis:
-    def __init__(self, experiments_data: List[AnalysisRBParams]) -> None:
+    def __init__(self, experiments_data: list[AnalysisRBParams]) -> None:
         """Init self.
 
         Args:
@@ -25,8 +25,8 @@ class RBAnalysis:
         self.experiments_data = experiments_data
         self._total_results: pd.DataFrame = pd.DataFrame()
 
-    async def _get_multiple_hardware_results_async(self) -> Dict[str, RbResults]:
-        total_result: Dict[str, RbResults] = {}
+    async def _get_multiple_hardware_results_async(self) -> dict[str, RbResults]:
+        total_result: dict[str, RbResults] = {}
         for batch in self.experiments_data:
             if len(batch.num_clifford) < 5:
                 raise ClassiqAnalyzerError(
@@ -38,7 +38,7 @@ class RBAnalysis:
         return total_result
 
     @staticmethod
-    def _get_df_indices(results: Dict[str, RbResults]) -> List[str]:
+    def _get_df_indices(results: dict[str, RbResults]) -> list[str]:
         temp_res = results.copy()
         _, rb_result_keys = temp_res.popitem()
         return list(rb_result_keys.__dict__.keys())
@@ -85,15 +85,15 @@ def _strict_string(arg: Union[PythonEnum, str]) -> str:
 
 
 def order_executor_data_by_hardware(
-    mixed_data: List[BackendPreferencesAndResult],
-) -> List[AnalysisRBParams]:
-    hardware_names: Set[str] = {
+    mixed_data: list[BackendPreferencesAndResult],
+) -> list[AnalysisRBParams]:
+    hardware_names: set[str] = {
         _strict_string(hardware.backend_name) for hardware, _, _ in mixed_data
     }
-    counts_dicts: Dict[str, List[Dict[str, int]]] = {
+    counts_dicts: dict[str, list[dict[str, int]]] = {
         name: list() for name in hardware_names
     }
-    cliffords_dicts: Dict[str, List[int]] = {name: list() for name in hardware_names}
+    cliffords_dicts: dict[str, list[int]] = {name: list() for name in hardware_names}
     for hardware, num_clifford, result in mixed_data:
         hw_name: str = _strict_string(hardware.backend_name)
         counts_dicts[hw_name].append(result.counts)  # type: ignore[union-attr]
@@ -110,7 +110,7 @@ def order_executor_data_by_hardware(
 
 
 def fit_to_exponential_function(
-    result: RbResults, num_clifford: List[int], ax: Optional[plt.Axes] = None
+    result: RbResults, num_clifford: list[int], ax: Optional[plt.Axes] = None
 ) -> None:
     if ax is None:
         plt.figure()

classiq/applications/__init__.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from classiq.applications import chemistry, combinatorial_optimization, finance, qsvm
 
 __all__ = [
@@ -13,5 +11,5 @@ __all__ = [
 _NON_IMPORTED_PUBLIC_SUBMODULES = ["qnn"]
 
 
-def __dir__() -> List[str]:
+def __dir__() -> list[str]:
     return __all__ + _NON_IMPORTED_PUBLIC_SUBMODULES

classiq/applications/chemistry/__init__.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from classiq.interface.chemistry.fermionic_operator import (
     FermionicOperator,
     SummedFermionicOperator,
@@ -38,5 +36,5 @@ __all__ = [
 ]
 
 
-def __dir__() -> List[str]:
+def __dir__() -> list[str]:
     return __all__

classiq/applications/chemistry/ansatz_parameters.py

@@ -1,5 +1,5 @@
 import dataclasses
-from typing import List, Tuple, Union
+from typing import Union
 
 from classiq.interface.generator.excitations import EXCITATIONS_TYPE_EXACT
 from classiq.interface.generator.ucc import default_excitation_factory
@@ -21,9 +21,9 @@ class HVAParameters:
 class HEAParameters:
     reps: int
     num_qubits: int
-    connectivity_map: List[Tuple[int, int]]
-    one_qubit_gates: List[str]
-    two_qubit_gates: List[str]
+    connectivity_map: list[tuple[int, int]]
+    one_qubit_gates: list[str]
+    two_qubit_gates: list[str]
 
 
 AnsatzParameters = Union[UCCParameters, HVAParameters, HEAParameters]