From 9acab20e131d32fd7f9dfbfdcd38a1a14b49663f Mon Sep 17 00:00:00 2001 From: Chris Sellers Date: Sun, 21 Apr 2024 09:03:23 +1000 Subject: [PATCH] Add type annotation --- nautilus_trader/persistence/catalog/parquet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nautilus_trader/persistence/catalog/parquet.py b/nautilus_trader/persistence/catalog/parquet.py index bd5567ce9182..d9793df276eb 100644 --- a/nautilus_trader/persistence/catalog/parquet.py +++ b/nautilus_trader/persistence/catalog/parquet.py @@ -337,7 +337,7 @@ def key(obj: Any) -> tuple[str, str | None]: return name, obj.instrument_id.value return name, None - def obj_to_type(obj) -> type: + def obj_to_type(obj: Data) -> type: return type(obj) if not isinstance(obj, CustomData) else obj.data.__class__ name_to_cls = {cls.__name__: cls for cls in {obj_to_type(d) for d in data}}