andras/great-ai
andras/great-ai
Watch
1
0
Fork
You've already forked great-ai
 0
Code Issues Pull requests 2 Projects Releases 4 Packages 1 Wiki Activity Actions

Ship the PEP 561 marker with the typed package

Browse source
This commit is contained in:
Andras Schmelczer 2026-10-27 20:04:31 +00:00
commit c5aa8406e1
3 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

1
great_ai/py.typed Normal file
View file

@ -0,0 +1 @@

6
great_ai/utilities/evaluate_ranking/evaluate_ranking.py
View file

@ -1,5 +1,5 @@
from pathlib import Path from pathlib import Path
from typing import Dict, List, Optional, TypeVar, cast from typing import Dict, Optional, Sequence, TypeVar, cast
import matplotlib import matplotlib
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
@ -14,8 +14,8 @@ T = TypeVar("T", str, float)
def evaluate_ranking( def evaluate_ranking(
expected: List[T], expected: Sequence[T],
actual_scores: List[float], actual_scores: Sequence[float],
target_recall: float, target_recall: float,
title: Optional[str] = "", title: Optional[str] = "",
disable_interpolation: bool = False, disable_interpolation: bool = False,

2
tests/test_prediction_regressions.py
View file

@ -99,7 +99,7 @@ def test_flattening_preserves_reserved_trace_fields() -> None:
def test_mongo_update_uses_valid_replacement() -> None: def test_mongo_update_uses_valid_replacement() -> None:
trace = Trace( trace: Trace = Trace(
trace_id="id", trace_id="id",
created="2026-01-01", created="2026-01-01",
original_execution_time_ms=0, original_execution_time_ms=0,