refactor: Add optional type hint for extra parameter in add_result_pair method

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-22 00:37:49 -03:00
commit cada4e8570

View file

@ -1,4 +1,4 @@
from typing import Any, List
from typing import Optional, Any, List
from pydantic import BaseModel
@ -13,7 +13,7 @@ class Payload(BaseModel):
def __iter__(self):
return iter(self.result_pairs)
def add_result_pair(self, result: Any, extra: Any = None) -> None:
def add_result_pair(self, result: Any, extra: Optional[Any] = None) -> None:
self.result_pairs.append(ResultPair(result=result, extra=extra))
def get_last_result_pair(self) -> ResultPair: