Fix type checking

coding-kitties · Mar 27, 2024 · 7f0a484 · 7f0a484
1 parent b2896aa
commit 7f0a484
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/resources/stubs/market_service_stub.py b/tests/resources/stubs/market_service_stub.py
@@ -1,3 +1,4 @@
+from typing import Dict
 from datetime import datetime
 from random import randint
 
@@ -33,11 +34,11 @@ def orders(self, value):
         self._orders = value
 
     @property
-    def balances(self) -> dict[str, float]:
+    def balances(self) -> Dict[str, float]:
         return self._balances
 
     @balances.setter
-    def balances(self, value: dict[str, float]):
+    def balances(self, value: Dict[str, float]):
         self._balances = value
 
     @property
@@ -150,7 +151,7 @@ def create_limit_sell_order(
             trading_symbol=trading_symbol,
         )
 
-    def get_balance(self, market) -> dict[str, float]:
+    def get_balance(self, market) -> Dict[str, float]:
         return self._balances
 
     def get_order(self, order, market):