Skip to content

Commit 2d951dd

Browse files
committed
fix lints; close #284
1 parent 4780b9e commit 2d951dd

File tree

3 files changed

+52
-2
lines changed

3 files changed

+52
-2
lines changed

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@ dev = [
6666
"proxy-py>=2.4.9",
6767
"sphinx-immaterial>=0.12.5",
6868
"mypy>=1.18.2",
69+
"pandas-stubs>=2.2.2.240807",
6970
]
7071

7172
[tool.setuptools.package-data]

tastytrade/utils.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from zoneinfo import ZoneInfo
77

88
from httpx import AsyncClient, Client, Response
9+
from pandas import Timestamp
910
from pandas_market_calendars import get_calendar # type: ignore[import-untyped]
1011
from pydantic import BaseModel, ConfigDict
1112

@@ -51,8 +52,8 @@ def is_market_open_now() -> bool:
5152
return False
5253

5354
# Use iloc[0] since schedule has only one row for a single day
54-
market_open = sched.iloc[0]["market_open"]
55-
market_close = sched.iloc[0]["market_close"]
55+
market_open: Timestamp = sched.iloc[0]["market_open"]
56+
market_close: Timestamp = sched.iloc[0]["market_close"]
5657
return market_open <= now_in_new_york() < market_close
5758

5859

uv.lock

Lines changed: 48 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)