diff --git a/backend/infrahub/workflows/models.py b/backend/infrahub/workflows/models.py index 7392167b6f..a17bc39e2b 100644 --- a/backend/infrahub/workflows/models.py +++ b/backend/infrahub/workflows/models.py @@ -1,5 +1,5 @@ import importlib -from typing import Any, Awaitable, Callable, Self, TypeVar +from typing import Any, Awaitable, Callable, TypeVar from uuid import UUID from prefect.client.orchestration import PrefectClient @@ -7,6 +7,7 @@ from prefect.client.schemas.objects import FlowRun from prefect.client.schemas.schedules import CronSchedule from pydantic import BaseModel +from typing_extensions import Self from infrahub import __version__ diff --git a/backend/tests/unit/git/test_git_rpc.py b/backend/tests/unit/git/test_git_rpc.py index 05bcb7bc99..a9e8438ae7 100644 --- a/backend/tests/unit/git/test_git_rpc.py +++ b/backend/tests/unit/git/test_git_rpc.py @@ -1,9 +1,10 @@ from __future__ import annotations -from typing import TYPE_CHECKING, Any, Optional, Self +from typing import TYPE_CHECKING, Any, Optional from unittest.mock import AsyncMock, patch from infrahub_sdk import UUIDT, Config, InfrahubClient +from typing_extensions import Self from infrahub.core.constants import InfrahubKind, RepositoryInternalStatus from infrahub.exceptions import RepositoryError