remove unnecessary casts

microsoft · Jan 16, 2025 · 89c73fe · 89c73fe
1 parent 7572964
commit 89c73fe
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/packages/autogen-core/src/autogen_core/_cache_store.py b/python/packages/autogen-core/src/autogen_core/_cache_store.py
@@ -1,4 +1,4 @@
-from typing import Any, Dict, Generic, Optional, Protocol, TypeVar, cast
+from typing import Dict, Generic, Optional, Protocol, TypeVar
 
 T = TypeVar("T")
 
@@ -37,10 +37,10 @@ def set(self, key: str, value: T) -> None:
 
 class InMemoryStore(CacheStore[T]):
     def __init__(self) -> None:
-        self.store: Dict[str, Any] = {}
+        self.store: Dict[str, T] = {}
 
     def get(self, key: str, default: Optional[T] = None) -> Optional[T]:
-        return cast(Optional[T], self.store.get(key, default))
+        return self.store.get(key, default)
 
     def set(self, key: str, value: T) -> None:
         self.store[key] = value