Skip to content

Commit e2b1c05

Browse files
rich7420shahar1
andauthored
fix mypy type errors in test_pool.py (apache#56801)
Co-authored-by: Shahar Epstein <60007259+shahar1@users.noreply.github.com>
1 parent d4d8d2f commit e2b1c05

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

airflow-core/tests/unit/models/test_pool.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,10 @@
1919

2020
from typing import TYPE_CHECKING
2121

22+
import pendulum
2223
import pytest
2324

2425
from airflow import settings
25-
from airflow._shared.timezones import timezone
2626
from airflow.exceptions import AirflowException, PoolNotFound
2727
from airflow.models.dag_version import DagVersion
2828
from airflow.models.pool import Pool
@@ -39,12 +39,13 @@
3939
)
4040

4141
if TYPE_CHECKING:
42+
from sqlalchemy.orm import Session
43+
4244
from airflow.models.team import Team
43-
from airflow.settings import Session
4445

4546
pytestmark = pytest.mark.db_test
4647

47-
DEFAULT_DATE = timezone.datetime(2016, 1, 1)
48+
DEFAULT_DATE = pendulum.datetime(2016, 1, 1, tz="UTC")
4849

4950

5051
class TestPool:

0 commit comments

Comments
 (0)