瀏覽代碼

fix type errors

raylu 1 月之前
父節點
當前提交
bcca6404a7
共有 1 個文件被更改,包括 3 次插入 和 3 次删除
  1. 3 3
      py/prepare.py

+ 3 - 3
py/prepare.py

@@ -62,7 +62,7 @@ def get_prices(f: typing.TextIO) -> typing.Mapping[str, float]:
 	prices.update(hardcoded_prices)
 	return prices
 
-def add_company_ranks(totals: dict[str, CompanyTotals]) -> dict[str, CompanyTotals]:
+def add_company_ranks(totals: typing.Mapping[str, CompanyTotals]) -> typing.Mapping[str, CompanyTotals]:
 	ranked = sorted(totals.items(), key=lambda item: item[1]['volume'], reverse=True)
 	for rank, (company_id, company_totals) in enumerate(ranked, start=1):
 		company_totals['volumeRank'] = rank
@@ -123,6 +123,6 @@ class CompanyTickerData(typing.TypedDict):
 	volume: float
 	rank: int
 
-class CompanyTotals(typing.TypedDict, total=False):
+class CompanyTotals(typing.TypedDict):
 	volume: float
-	volumeRank: int
+	volumeRank: typing.NotRequired[int]