From 39337e4ede9555708e0f333251b41145095c669b Mon Sep 17 00:00:00 2001 From: fubuloubu <3859395+fubuloubu@users.noreply.github.com> Date: Tue, 16 Apr 2024 22:55:20 -0400 Subject: [PATCH] fix: use Dict from typing for 3.8 compat --- silverback/middlewares.py | 4 ++-- silverback/recorder.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/silverback/middlewares.py b/silverback/middlewares.py index 02ab15e0..ba09eff4 100644 --- a/silverback/middlewares.py +++ b/silverback/middlewares.py @@ -1,4 +1,4 @@ -from typing import Any +from typing import Any, Dict from ape.logging import logger from ape.types import ContractLog @@ -26,7 +26,7 @@ def compute_block_time() -> int: def pre_send(self, message: TaskiqMessage) -> TaskiqMessage: # TODO: Necessary because bytes/HexBytes doesn't encode/deocde well for some reason def fix_dict(data: dict, recurse_count: int = 0) -> dict: - fixed_data: dict[str, Any] = {} + fixed_data: Dict[str, Any] = {} for name, value in data.items(): if isinstance(value, bytes): fixed_data[name] = to_hex(value) diff --git a/silverback/recorder.py b/silverback/recorder.py index 89754af9..173dc4b3 100644 --- a/silverback/recorder.py +++ b/silverback/recorder.py @@ -1,6 +1,6 @@ from abc import ABC, abstractmethod from pathlib import Path -from typing import Any, Iterator, Optional +from typing import Any, Dict, Iterator, Optional from ape.logging import get_logger from pydantic import BaseModel, Field @@ -36,10 +36,10 @@ class TaskResult(BaseModel): block_number: Optional[int] = None # Custom user metrics here - metrics: dict[str, Datapoint] = {} + metrics: Dict[str, Datapoint] = {} @classmethod - def _extract_custom_metrics(cls, result: Any, task_name: str) -> dict[str, Datapoint]: + def _extract_custom_metrics(cls, result: Any, task_name: str) -> Dict[str, Datapoint]: if isinstance(result, Datapoint): # type: ignore[arg-type,misc] return {"result": result}