diff --git a/metadata-ingestion/src/datahub/ingestion/source/kafka.py b/metadata-ingestion/src/datahub/ingestion/source/kafka.py index df6722bf8bcb5c..13e01e8e038907 100644 --- a/metadata-ingestion/src/datahub/ingestion/source/kafka.py +++ b/metadata-ingestion/src/datahub/ingestion/source/kafka.py @@ -12,7 +12,6 @@ AdminClient, ConfigEntry, ConfigResource, - ResourceType, TopicMetadata, ) @@ -404,11 +403,10 @@ def fetch_extra_topic_details(self, topics: List[str]) -> Dict[str, dict]: def fetch_topic_configurations(self, topics: List[str]) -> Dict[str, dict]: logger.info("Fetching config details for all topics") - configs: Dict[ ConfigResource, concurrent.futures.Future ] = self.admin_client.describe_configs( - resources=[ConfigResource(ResourceType.TOPIC, t) for t in topics], + resources=[ConfigResource(ConfigResource.Type.TOPIC, t) for t in topics], request_timeout=self.source_config.connection.client_timeout_seconds, ) logger.debug("Waiting for config details futures to complete")