classiq 0.93.0__py3-none-any.whl → 0.94.0__py3-none-any.whl

classiq/evaluators/qmod_type_inference/classical_type_inference.py

@@ -1,5 +1,5 @@
 from enum import IntEnum
-from typing import Any, Optional, Union
+from typing import Any
 
 import sympy
 
@@ -18,14 +18,13 @@ from classiq.interface.generator.functions.classical_type import (
 )
 from classiq.interface.generator.functions.type_name import Enum, Struct, TypeName
 from classiq.interface.generator.types.struct_declaration import StructDeclaration
-from classiq.interface.helpers.backward_compatibility import zip_strict
 
 from classiq.evaluators.qmod_node_evaluators.utils import get_sympy_type
 
 
 def _copy_generative_flag(
-    from_type: ClassicalType, to_type: Optional[ClassicalType]
-) -> Optional[ClassicalType]:
+    from_type: ClassicalType, to_type: ClassicalType | None
+) -> ClassicalType | None:
     if to_type is None:
         return None
     if from_type.is_generative:
@@ -65,11 +64,11 @@ def infer_classical_type(value: Any) -> ClassicalType:
 
 
 def _inject_classical_array_attributes(
-    from_type: ClassicalType, to_type: Union[ClassicalArray, ClassicalTuple]
-) -> Optional[ClassicalType]:
+    from_type: ClassicalType, to_type: ClassicalArray | ClassicalTuple
+) -> ClassicalType | None:
     if isinstance(to_type, ClassicalArray):
         if isinstance(from_type, ClassicalArray):
-            length: Optional[Expression]
+            length: Expression | None
             if from_type.has_constant_length:
                 if (
                     to_type.has_constant_length
@@ -116,7 +115,7 @@ def _inject_classical_array_attributes(
             return None
         element_types = [
             inject_classical_type_attributes(from_element_type, to_element_type)
-            for from_element_type, to_element_type in zip_strict(
+            for from_element_type, to_element_type in zip(
                 from_type.element_types, to_type.element_types, strict=True
             )
         ]
@@ -128,7 +127,7 @@ def _inject_classical_array_attributes(
 
 def _inject_classical_type_name_attributes(
     from_type: ClassicalType, to_type: TypeName
-) -> Optional[ClassicalType]:
+) -> ClassicalType | None:
     if to_type.is_enum:
         if isinstance(from_type, Integer) or (
             isinstance(from_type, TypeName) and from_type.name == to_type.name
@@ -157,7 +156,7 @@ def _inject_classical_type_name_attributes(
 
 def inject_classical_type_attributes(
     from_type: ClassicalType, to_type: ClassicalType
-) -> Optional[ClassicalType]:
+) -> ClassicalType | None:
     if isinstance(to_type, Bool):
         if isinstance(from_type, Bool):
             return _copy_generative_flag(to_type, Bool())

classiq/evaluators/qmod_type_inference/quantum_type_inference.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Optional
+from typing import TYPE_CHECKING
 
 from classiq.interface.exceptions import (
     ClassiqExpansionError,
@@ -80,7 +80,7 @@ def _same_shape(quantum_type_1: QuantumType, quantum_type_2: QuantumType) -> boo
 def _inject_qnum_type_attributes(
     from_type: QuantumType, to_type: QuantumNumeric
 ) -> QuantumNumeric:
-    size: Optional[Expression]
+    size: Expression | None
     if from_type.has_size_in_bits:
         size = Expression(expr=str(from_type.size_in_bits))
     else:
@@ -126,7 +126,7 @@ def _inject_qnum_type_attributes(
 
 def _inject_qarray_type_attributes(
     from_type: QuantumType, to_type: QuantumBitvector
-) -> Optional[QuantumBitvector]:
+) -> QuantumBitvector | None:
     if _same_shape(from_type, to_type):
         if TYPE_CHECKING:
             assert isinstance(from_type, QuantumBitvector)
@@ -182,7 +182,7 @@ def _inject_qarray_type_attributes(
 
 def _inject_qstruct_type_attributes(
     from_type: QuantumType, to_type: TypeName
-) -> Optional[TypeName]:
+) -> TypeName | None:
     if isinstance(from_type, TypeName) and from_type.name == to_type.name:
         fields = {
             field_name: inject_quantum_type_attributes(
@@ -241,7 +241,7 @@ def _inject_qstruct_type_attributes(
 
 def inject_quantum_type_attributes(
     from_type: QuantumType, to_type: QuantumType
-) -> Optional[QuantumType]:
+) -> QuantumType | None:
     for qmod_type in (from_type, to_type):
         if isinstance(qmod_type, TypeName) and not qmod_type.has_fields:
             raise ClassiqInternalExpansionError

classiq/execution/execution_session.py

@@ -1,8 +1,9 @@
 import inspect
 import random
 import warnings
+from collections.abc import Callable
 from types import TracebackType
-from typing import Any, Callable, Optional, Union, cast
+from typing import Any, Optional, Union, cast
 
 from classiq.interface.chemistry.operator import PauliOperator, pauli_integers_to_str
 from classiq.interface.exceptions import (
@@ -104,7 +105,7 @@ class ExecutionSession:
     def __init__(
         self,
         quantum_program: QuantumProgram,
-        execution_preferences: Optional[ExecutionPreferences] = None,
+        execution_preferences: ExecutionPreferences | None = None,
     ):
         self.program: QuantumProgram = quantum_program
         self.update_execution_preferences(execution_preferences)
@@ -124,9 +125,9 @@ class ExecutionSession:
 
     def __exit__(
         self,
-        exc_type: Optional[type[BaseException]],
-        exc_val: Optional[BaseException],
-        exc_tb: Optional[TracebackType],
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        exc_tb: TracebackType | None,
     ) -> None:
         self.close()
 
@@ -156,7 +157,7 @@ class ExecutionSession:
         return ExecutionJob(details=result)
 
     def update_execution_preferences(
-        self, execution_preferences: Optional[ExecutionPreferences]
+        self, execution_preferences: ExecutionPreferences | None
     ) -> None:
         """
         Update the execution preferences for the session.
@@ -170,7 +171,7 @@ class ExecutionSession:
         if execution_preferences is not None:
             self.program.model.execution_preferences = execution_preferences
 
-    def sample(self, parameters: Optional[ExecutionParams] = None) -> ExecutionDetails:
+    def sample(self, parameters: ExecutionParams | None = None) -> ExecutionDetails:
         """
         Samples the quantum program with the given parameters, if any.
 
@@ -183,9 +184,7 @@ class ExecutionSession:
         job = self.submit_sample(parameters=parameters)
         return job.get_sample_result(_http_client=self._async_client)
 
-    def submit_sample(
-        self, parameters: Optional[ExecutionParams] = None
-    ) -> ExecutionJob:
+    def submit_sample(self, parameters: ExecutionParams | None = None) -> ExecutionJob:
         """
         Initiates an execution job with the `sample` primitive.
 
@@ -235,7 +234,7 @@ class ExecutionSession:
         return self._execute(execution_primitives_input)
 
     def estimate(
-        self, hamiltonian: Hamiltonian, parameters: Optional[ExecutionParams] = None
+        self, hamiltonian: Hamiltonian, parameters: ExecutionParams | None = None
     ) -> EstimationResult:
         """
         Estimates the expectation value of the given Hamiltonian using the quantum program.
@@ -256,7 +255,7 @@ class ExecutionSession:
     def submit_estimate(
         self,
         hamiltonian: Hamiltonian,
-        parameters: Optional[ExecutionParams] = None,
+        parameters: ExecutionParams | None = None,
         *,
         _check_deprecation: bool = True,
     ) -> ExecutionJob:
@@ -336,11 +335,11 @@ class ExecutionSession:
 
     def minimize(
         self,
-        cost_function: Union[Hamiltonian, QmodExpressionCreator],
+        cost_function: Hamiltonian | QmodExpressionCreator,
         initial_params: ExecutionParams,
         max_iteration: int,
         quantile: float = 1.0,
-        tolerance: Optional[float] = None,
+        tolerance: float | None = None,
     ) -> list[tuple[float, ExecutionParams]]:
         """
         Minimizes the given cost function using the quantum program.
@@ -378,11 +377,11 @@ class ExecutionSession:
 
     def submit_minimize(
         self,
-        cost_function: Union[Hamiltonian, QmodExpressionCreator],
+        cost_function: Hamiltonian | QmodExpressionCreator,
         initial_params: ExecutionParams,
         max_iteration: int,
         quantile: float = 1.0,
-        tolerance: Optional[float] = None,
+        tolerance: float | None = None,
         *,
         _check_deprecation: bool = True,
     ) -> ExecutionJob:
@@ -416,9 +415,9 @@ class ExecutionSession:
                 "The initial parameters must be a dictionary with a single key-value pair."
             )
 
-        _cost_function: Union[PauliOperator, Expression]
+        _cost_function: PauliOperator | Expression
         _initial_params = parse_params(initial_params)
-        minimize: Union[MinimizeQuantumCostInput, MinimizeClassicalCostInput]
+        minimize: MinimizeQuantumCostInput | MinimizeClassicalCostInput
         if callable(cost_function):
             circuit_output_types = self.program.model.circuit_output_types
             _cost_function = self._create_qmod_expression(
@@ -447,7 +446,7 @@ class ExecutionSession:
     def estimate_cost(
         self,
         cost_func: Callable[[ParsedState], float],
-        parameters: Optional[ExecutionParams] = None,
+        parameters: ExecutionParams | None = None,
         quantile: float = 1.0,
     ) -> float:
         """

classiq/execution/jobs.py

@@ -1,7 +1,7 @@
 import warnings
 import webbrowser
 from datetime import datetime
-from typing import Any, Optional, Union
+from typing import Any
 from urllib.parse import urljoin
 
 import httpx
@@ -39,7 +39,7 @@ class ClassiqExecutionResultError(ClassiqError):
 
 class ExecutionJob:
     _details: ExecutionJobDetails
-    _result: Optional[ResultsCollection]
+    _result: ResultsCollection | None
 
     def __init__(self, details: ExecutionJobDetails) -> None:
         self._details = details
@@ -50,7 +50,7 @@ class ExecutionJob:
         return self._details.id
 
     @property
-    def name(self) -> Optional[str]:
+    def name(self) -> str | None:
         return self._details.name
 
     @property
@@ -58,15 +58,15 @@ class ExecutionJob:
         return self._details.start_time
 
     @property
-    def end_time(self) -> Optional[datetime]:
+    def end_time(self) -> datetime | None:
         return self._details.end_time
 
     @property
-    def provider(self) -> Optional[str]:
+    def provider(self) -> str | None:
         return self._details.provider
 
     @property
-    def backend_name(self) -> Optional[str]:
+    def backend_name(self) -> str | None:
         return self._details.backend_name
 
     @property
@@ -74,15 +74,15 @@ class ExecutionJob:
         return self._details.status
 
     @property
-    def num_shots(self) -> Optional[int]:
+    def num_shots(self) -> int | None:
         return self._details.num_shots
 
     @property
-    def program_id(self) -> Optional[str]:
+    def program_id(self) -> str | None:
         return self._details.program_id
 
     @property
-    def error(self) -> Optional[str]:
+    def error(self) -> str | None:
         return self._details.error
 
     def __repr__(self) -> str:
@@ -92,7 +92,7 @@ class ExecutionJob:
         else:
             return f"{class_name}(name={self.name!r}, id={self.id!r})"
 
-    def cost(self, *, verbose: bool = False) -> Union[str, JobCost]:
+    def cost(self, *, verbose: bool = False) -> str | JobCost:
         if self._details.cost is None:
             self._details.cost = JobCost()
         if verbose:
@@ -103,7 +103,7 @@ class ExecutionJob:
     async def from_id_async(
         cls,
         id: str,
-        _http_client: Optional[httpx.AsyncClient] = None,
+        _http_client: httpx.AsyncClient | None = None,
     ) -> "ExecutionJob":
         details = await ApiWrapper.call_get_execution_job_details(
             JobID(job_id=id), http_client=_http_client
@@ -116,7 +116,7 @@ class ExecutionJob:
     def from_id(
         cls,
         id: str,
-        _http_client: Optional[httpx.AsyncClient] = None,
+        _http_client: httpx.AsyncClient | None = None,
     ) -> "ExecutionJob":
         return syncify_function(cls.from_id_async)(id, _http_client=_http_client)
 
@@ -126,8 +126,8 @@ class ExecutionJob:
 
     async def result_async(
         self,
-        timeout_sec: Optional[float] = None,
-        _http_client: Optional[httpx.AsyncClient] = None,
+        timeout_sec: float | None = None,
+        _http_client: httpx.AsyncClient | None = None,
     ) -> ResultsCollection:
         await self.poll_async(timeout_sec=timeout_sec, _http_client=_http_client)
 
@@ -151,7 +151,7 @@ class ExecutionJob:
         return self.result(*args, **kwargs)[0].value
 
     def get_sample_result(
-        self, _http_client: Optional[httpx.AsyncClient] = None
+        self, _http_client: httpx.AsyncClient | None = None
     ) -> ExecutionDetails:
         """
         Returns the job's result as a single sample result after validation. If the result is not yet available, waits for it.
@@ -177,7 +177,7 @@ class ExecutionJob:
         return result
 
     def get_batch_sample_result(
-        self, _http_client: Optional[httpx.AsyncClient] = None
+        self, _http_client: httpx.AsyncClient | None = None
     ) -> list[ExecutionDetails]:
         """
         Returns the job's result as a single batch_sample result after validation. If the result is not yet available, waits for it.
@@ -209,7 +209,7 @@ class ExecutionJob:
         return result_list
 
     def get_estimate_result(
-        self, _http_client: Optional[httpx.AsyncClient] = None
+        self, _http_client: httpx.AsyncClient | None = None
     ) -> EstimationResult:
         """
         Returns the job's result as a single estimate result after validation. If the result is not yet available, waits for it.
@@ -233,7 +233,7 @@ class ExecutionJob:
         raise ClassiqExecutionResultError("estimate")
 
     def get_batch_estimate_result(
-        self, _http_client: Optional[httpx.AsyncClient] = None
+        self, _http_client: httpx.AsyncClient | None = None
     ) -> list[EstimationResult]:
         """
         Returns the job's result as a single batch_estimate result after validation. If the result is not yet available, waits for it.
@@ -258,7 +258,7 @@ class ExecutionJob:
         raise ClassiqExecutionResultError("batch_estimate")
 
     def get_minimization_result(
-        self, _http_client: Optional[httpx.AsyncClient] = None
+        self, _http_client: httpx.AsyncClient | None = None
     ) -> TaggedMinimizeResult:
         """
         Returns the job's result as a single minimization result after validation. If the result is not yet available, waits for it.
@@ -281,8 +281,8 @@ class ExecutionJob:
 
     async def poll_async(
         self,
-        timeout_sec: Optional[float] = None,
-        _http_client: Optional[httpx.AsyncClient] = None,
+        timeout_sec: float | None = None,
+        _http_client: httpx.AsyncClient | None = None,
     ) -> None:
         if not self.status.is_final():
             await self._poll_job(timeout_sec=timeout_sec, _http_client=_http_client)
@@ -291,10 +291,10 @@ class ExecutionJob:
 
     async def _poll_job(
         self,
-        timeout_sec: Optional[float] = None,
-        _http_client: Optional[httpx.AsyncClient] = None,
+        timeout_sec: float | None = None,
+        _http_client: httpx.AsyncClient | None = None,
     ) -> None:
-        def response_parser(json_response: JSONObject) -> Optional[bool]:
+        def response_parser(json_response: JSONObject) -> bool | None:
             self._details = ExecutionJobDetails.model_validate(json_response)
             if self.status.is_final():
                 return True
@@ -313,7 +313,7 @@ class ExecutionJob:
     async def rename_async(
         self,
         name: str,
-        _http_client: Optional[httpx.AsyncClient] = None,
+        _http_client: httpx.AsyncClient | None = None,
     ) -> None:
         self._details = await ApiWrapper.call_patch_execution_job(
             self._job_id,
@@ -325,7 +325,7 @@ class ExecutionJob:
 
     async def cancel_async(
         self,
-        _http_client: Optional[httpx.AsyncClient] = None,
+        _http_client: httpx.AsyncClient | None = None,
     ) -> None:
         """
         Cancels the execution job. This implies the cancellation of any ongoing jobs

classiq/execution/qnn.py

@@ -1,5 +1,4 @@
 import functools
-from typing import Optional
 
 import more_itertools
 
@@ -60,7 +59,7 @@ def _execute_qnn_sample(
 def execute_qnn(
     quantum_program: QuantumProgram,
     arguments: MultipleArguments,
-    observable: Optional[PauliOperator] = None,
+    observable: PauliOperator | None = None,
 ) -> ResultsCollection:
     with ExecutionSession(quantum_program) as session:
         if observable:

classiq/execution/user_budgets.py

@@ -1,10 +1,8 @@
-from typing import Optional
-
 from classiq.interface.backend.quantum_backend_providers import ProviderVendor
 from classiq.interface.executor.user_budget import UserBudgets
 
+from classiq._internals import async_utils
 from classiq._internals.api_wrapper import ApiWrapper
-from classiq._internals.async_utils import syncify_function
 
 PROVIDER_MAPPER = {
     ProviderVendor.IONQ: "IONQ",
@@ -22,7 +20,7 @@ PROVIDER_MAPPER = {
 
 
 async def get_budget_async(
-    provider_vendor: Optional[ProviderVendor] = None,
+    provider_vendor: ProviderVendor | None = None,
 ) -> UserBudgets:
 
     budgets_list = await ApiWrapper().call_get_all_budgets()
@@ -35,7 +33,21 @@ async def get_budget_async(
     return UserBudgets(budgets=budgets_list)
 
 
-get_budget = syncify_function(get_budget_async)
+def get_budget(
+    provider_vendor: ProviderVendor | None = None,
+) -> UserBudgets:
+    """
+    Retrieve the user's budget information for quantum computing resources.
+
+    Args:
+        provider_vendor:
+            (Optional) The quantum backend provider to filter budgets by.
+            If not provided, budgets for all providers will be returned.
+
+    Returns:
+        UserBudgets: An object containing the user's budget information.
+    """
+    return async_utils.run(get_budget_async(provider_vendor))
 
 
 async def set_budget_limit_async(
@@ -62,7 +74,26 @@ async def set_budget_limit_async(
     return UserBudgets(budgets=[budgets_list])
 
 
-set_budget_limit = syncify_function(set_budget_limit_async)
+def set_budget_limit(
+    provider_vendor: ProviderVendor,
+    limit: float,
+) -> UserBudgets:
+    """
+    Set a budget limit for a specific quantum backend provider.
+
+    Args:
+        provider_vendor:
+            The quantum backend provider for which to set the budget limit.
+        limit:
+            The budget limit to set. Must be greater than zero and not exceed the available budget.
+
+    Returns:
+        UserBudgets: An object containing the updated budget information.
+
+    Raises:
+        ValueError: If the provider is unsupported, no budget is found, or the limit is invalid.
+    """
+    return async_utils.run(set_budget_limit_async(provider_vendor, limit))
 
 
 async def clear_budget_limit_async(provider_vendor: ProviderVendor) -> UserBudgets:
@@ -74,4 +105,18 @@ async def clear_budget_limit_async(provider_vendor: ProviderVendor) -> UserBudge
     return UserBudgets(budgets=[budgets_list])
 
 
-clear_budget_limit = syncify_function(clear_budget_limit_async)
+def clear_budget_limit(provider_vendor: ProviderVendor) -> UserBudgets:
+    """
+    Clear the budget limit for a specific quantum backend provider.
+
+    Args:
+        provider_vendor:
+            The quantum backend provider for which to clear the budget limit.
+
+    Returns:
+        UserBudgets: An object containing the updated budget information.
+
+    Raises:
+        ValueError: If the provider is unsupported.
+    """
+    return async_utils.run(clear_budget_limit_async(provider_vendor))

classiq/executor.py

@@ -1,8 +1,6 @@
 """Executor module, implementing facilities for executing quantum programs using Classiq platform."""
 
-from typing import Union
-
-from typing_extensions import TypeAlias
+from typing import TypeAlias, Union
 
 from classiq.interface.backend.backend_preferences import BackendPreferencesTypes
 from classiq.interface.executor.estimation import OperatorsEstimation

classiq/interface/_version.py

@@ -3,5 +3,5 @@ from packaging.version import Version
 # This file was generated automatically
 # Please don't track in version control (DONTTRACK)
 
-SEMVER_VERSION = '0.93.0'
+SEMVER_VERSION = '0.94.0'
 VERSION = str(Version(SEMVER_VERSION))

classiq/interface/analyzer/analysis_params.py

@@ -1,4 +1,4 @@
-from typing import Annotated, Optional
+from typing import Annotated
 
 import pydantic
 from pydantic import Field, StringConstraints
@@ -26,7 +26,7 @@ class AnalysisParams(pydantic.BaseModel):
 
 
 class HardwareListParams(pydantic.BaseModel):
-    devices: Optional[list[PydanticNonEmptyString]] = pydantic.Field(
+    devices: list[PydanticNonEmptyString] | None = pydantic.Field(
         default=None, description="Devices"
     )
     providers: list[Provider]
@@ -35,7 +35,7 @@ class HardwareListParams(pydantic.BaseModel):
     @pydantic.field_validator("providers")
     @classmethod
     def set_default_providers(
-        cls, providers: Optional[list[AnalyzerProviderVendor]]
+        cls, providers: list[AnalyzerProviderVendor] | None
     ) -> list[AnalyzerProviderVendor]:
         if providers is None:
             providers = list(AnalyzerProviderVendor)
@@ -67,7 +67,7 @@ class LatexParams(AnalysisParams):
 
 
 class AnalysisHardwareTranspilationParams(pydantic.BaseModel):
-    hardware_data: Optional[SynthesisHardwareData] = None
+    hardware_data: SynthesisHardwareData | None = None
     random_seed: int
     transpilation_option: TranspilationOption
 

classiq/interface/analyzer/cytoscape_graph.py

@@ -1,4 +1,4 @@
-from typing import Any, Optional
+from typing import Any
 
 import pydantic
 
@@ -42,7 +42,7 @@ class CytoScapeNode(pydantic.BaseModel):
         default=...,
         description="Data of the Node, such as label, and color, can be of free form",
     )
-    position: Optional[CytoScapePosition] = pydantic.Field(
+    position: CytoScapePosition | None = pydantic.Field(
         default=..., description="Position of the Node to be rendered in Cytocape"
     )
 
@@ -66,7 +66,7 @@ class ConnectivityErrors(StrEnum):
 
 
 class HardwareConnectivityGraphResult(VersionedModel):
-    graph: Optional[CytoScapeGraph] = pydantic.Field(
+    graph: CytoScapeGraph | None = pydantic.Field(
         default=...,
         description="The Cytoscape graph in the desired Structure for the FE",
     )

classiq/interface/analyzer/result.py

@@ -1,4 +1,4 @@
-from typing import Annotated, Literal, Optional, Union
+from typing import Annotated, Literal, Union
 
 import pydantic
 from pydantic import Field
@@ -125,15 +125,15 @@ class HardwareComparisonDataColumns(pydantic.BaseModel):
 
 
 class AvailableHardware(pydantic.BaseModel):
-    ibm_quantum: Optional[dict[PydanticNonEmptyString, bool]] = pydantic.Field(
+    ibm_quantum: dict[PydanticNonEmptyString, bool] | None = pydantic.Field(
         default=None,
         description="available IBM Quantum devices with boolean indicates if a given device has enough qubits.",
     )
-    azure_quantum: Optional[dict[PydanticNonEmptyString, bool]] = pydantic.Field(
+    azure_quantum: dict[PydanticNonEmptyString, bool] | None = pydantic.Field(
         default=None,
         description="available Azure Quantum devices with boolean indicates if a given device has enough qubits.",
     )
-    amazon_braket: Optional[dict[PydanticNonEmptyString, bool]] = pydantic.Field(
+    amazon_braket: dict[PydanticNonEmptyString, bool] | None = pydantic.Field(
         default=None,
         description="available Amazon Braket devices with boolean indicates if a given device has enough qubits.",
     )

classiq/interface/applications/qsvm.py

@@ -1,7 +1,6 @@
 from collections.abc import Iterable as IterableType, Sequence
 from typing import (
     Any,
-    Optional,
     Union,
 )
 
@@ -16,7 +15,7 @@ DataList = list[list[float]]
 LabelsInt = list[int]
 
 
-def listify(obj: Union[IterableType, ArrayLike]) -> list:
+def listify(obj: IterableType | ArrayLike) -> list:
     if isinstance(obj, np.ndarray):
         return obj.tolist()
     elif isinstance(obj, Sequence) and obj and isinstance(obj[0], np.ndarray):
@@ -85,20 +84,18 @@ class QSVMInternalState(VersionedModel):
 
 class QSVMData(VersionedModel):
     data: DataList
-    labels: Optional[LabelsInt] = None
-    internal_state: Optional[QSVMInternalState] = None
+    labels: LabelsInt | None = None
+    internal_state: QSVMInternalState | None = None
     model_config = ConfigDict(extra="forbid")
 
     @pydantic.field_validator("data", mode="before")
     @classmethod
-    def set_data(cls, data: Union[IterableType, ArrayLike]) -> list:
+    def set_data(cls, data: IterableType | ArrayLike) -> list:
         return listify(data)
 
     @pydantic.field_validator("labels", mode="before")
     @classmethod
-    def set_labels(
-        cls, labels: Optional[Union[IterableType, ArrayLike]]
-    ) -> Optional[list]:
+    def set_labels(cls, labels: IterableType | ArrayLike | None) -> list | None:
         if labels is None:
             return None
         else: