Skip to content

Commit

Permalink
🏷️ (context) tweak types
Browse files Browse the repository at this point in the history
  • Loading branch information
simonwoerpel committed Jan 4, 2024
1 parent 41fe9e3 commit f213738
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions investigraph/model/context.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,13 @@
from datetime import datetime
from logging import Logger, LoggerAdapter
from typing import Iterable

from followthemoney.util import make_entity_id
from ftmq.io import smart_write
from ftmq.util import join_slug
from nomenklatura.entity import CE
from prefect import get_run_logger
from prefect.logging.loggers import MissingContextError, PrefectLogAdapter
from prefect.exceptions import MissingContextError
from pydantic import BaseModel

from investigraph.cache import Cache, get_cache
Expand Down Expand Up @@ -34,7 +35,7 @@ def cache(self) -> Cache:
return get_cache()

@property
def log(self) -> PrefectLogAdapter:
def log(self) -> LoggerAdapter | Logger:
try:
return get_run_logger()
except MissingContextError:
Expand Down

0 comments on commit f213738

Please sign in to comment.