diff options
Diffstat (limited to 'python/fatcat_tools/workers/worker_common.py')
| -rw-r--r-- | python/fatcat_tools/workers/worker_common.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/python/fatcat_tools/workers/worker_common.py b/python/fatcat_tools/workers/worker_common.py index 5239465b..f6b0293e 100644 --- a/python/fatcat_tools/workers/worker_common.py +++ b/python/fatcat_tools/workers/worker_common.py @@ -1,7 +1,7 @@ from typing import Any, Dict, Optional from confluent_kafka import Consumer, KafkaException, Message, TopicPartition -from fatcat_openapi_client import ApiClient +from fatcat_openapi_client import DefaultApi def most_recent_message(topic: str, kafka_config: Dict[str, Any]) -> Message: @@ -58,7 +58,7 @@ class FatcatWorker: kafka_hosts: str, produce_topic: Optional[str] = None, consume_topic: Optional[str] = None, - api: Optional[ApiClient] = None, + api: Optional[DefaultApi] = None, ) -> None: if api: self.api = api |