chore: Add ty for type checking (#24)

* Add ty to type checking

* clean up

* sync all groups

* sync all extras

Author: khrapovs

.github/workflows/workflow.yaml

@@ -84,9 +84,11 @@ jobs:
         with:
           python-version: "3.13"
       - name: Install dependencies
-        run: uv sync --group dev
+        run: uv sync --all-groups --all-extras
       - name: Run pre-commit
         run: uv run prek run -v --show-diff-on-failure
+      - name: Run ty
+        run: uv run ty check
 
   docs:
     name: Documentation

.gitignore

@@ -7,6 +7,7 @@ benchmark.json
 benchmark_history.svg
 __pycache__/
 venv/
+.venv/
 site/
 docs/reference/
 *.egg-info

pyproject.toml

@@ -31,6 +31,7 @@ test = [
 ]
 dev = [
     "prek>=0.2.30",
+    "ty>=0.0.12",
 ]
 [project.optional-dependencies]
 doc = [
@@ -84,6 +85,9 @@ disallow_untyped_calls = true
 disallow_untyped_defs = true
 disallow_incomplete_defs = true
 
+[tool.ty.rules]
+no-matching-overload = "ignore"
+
 [tool.commitizen]
 version_scheme = "semver"
 version_provider = "uv"

src/order_matching/executed_trades.py

@@ -3,9 +3,10 @@
 from collections import defaultdict
 from dataclasses import asdict
 from datetime import datetime
+from typing import cast
 
 import pandas as pd
-from pandera.typing import DataFrame
+from pandera.typing.pandas import DataFrame
 
 from order_matching.schemas import TradeDataSchema
 from order_matching.trade import Trade
@@ -66,13 +67,16 @@ def to_frame(self) -> DataFrame[TradeDataSchema]:
         """
         trades = self.trades
         if len(trades) == 0:
-            return pd.DataFrame()
+            return cast(DataFrame[TradeDataSchema], pd.DataFrame())
         else:
-            return pd.DataFrame.from_records([asdict(trade) for trade in trades]).assign(
-                **{
-                    TradeDataSchema.side: lambda df: df[TradeDataSchema.side].astype(str),
-                    TradeDataSchema.execution: lambda df: df[TradeDataSchema.execution].astype(str),
-                }
+            return cast(
+                DataFrame[TradeDataSchema],
+                pd.DataFrame.from_records([asdict(trade) for trade in trades]).assign(
+                    **{
+                        TradeDataSchema.side: lambda df: df[TradeDataSchema.side].astype(str),
+                        TradeDataSchema.execution: lambda df: df[TradeDataSchema.execution].astype(str),
+                    }
+                ),
             )
 
     def __add__(self, other: ExecutedTrades) -> ExecutedTrades:

src/order_matching/orders.py

@@ -1,10 +1,10 @@
 from __future__ import annotations
 
 from dataclasses import asdict
-from typing import Generator, Iterator, Sequence
+from typing import Generator, Iterator, Sequence, cast
 
 import pandas as pd
-from pandera.typing import DataFrame
+from pandera.typing.pandas import DataFrame
 
 from order_matching.order import Order
 from order_matching.schemas import OrderDataSchema
@@ -63,20 +63,23 @@ def to_frame(self) -> DataFrame[OrderDataSchema]:
         DataFrame[OrderDataSchema]
         """
         if len(self.orders) == 0:
-            return pd.DataFrame()
+            return cast(DataFrame[OrderDataSchema], pd.DataFrame())
         else:
-            return pd.DataFrame.from_records([asdict(order) for order in self.orders]).assign(
-                **{
-                    OrderDataSchema.side: lambda df: df[OrderDataSchema.side].astype(str),
-                    OrderDataSchema.execution: lambda df: df[OrderDataSchema.execution].astype(str),
-                    OrderDataSchema.status: lambda df: df[OrderDataSchema.status].astype(str),
-                    OrderDataSchema.timestamp: lambda df: pd.to_datetime(
-                        df[OrderDataSchema.timestamp], errors="coerce"
-                    ),
-                    OrderDataSchema.expiration: lambda df: pd.to_datetime(
-                        df[OrderDataSchema.expiration], errors="coerce"
-                    ),
-                }
+            return cast(
+                DataFrame[OrderDataSchema],
+                pd.DataFrame.from_records([asdict(order) for order in self.orders]).assign(
+                    **{
+                        OrderDataSchema.side: lambda df: df[OrderDataSchema.side].astype(str),
+                        OrderDataSchema.execution: lambda df: df[OrderDataSchema.execution].astype(str),
+                        OrderDataSchema.status: lambda df: df[OrderDataSchema.status].astype(str),
+                        OrderDataSchema.timestamp: lambda df: pd.to_datetime(
+                            df[OrderDataSchema.timestamp], errors="coerce"
+                        ),
+                        OrderDataSchema.expiration: lambda df: pd.to_datetime(
+                            df[OrderDataSchema.expiration], errors="coerce"
+                        ),
+                    }
+                ),
             )
 
     @property