Skip to content

Commit 73e26fc

Browse files
committed
Fix pydantic issue
1 parent 724d682 commit 73e26fc

File tree

1 file changed

+3
-2
lines changed
  • metadata-ingestion/src/datahub/ingestion/source/kafka_connect

1 file changed

+3
-2
lines changed

metadata-ingestion/src/datahub/ingestion/source/kafka_connect/common.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from dataclasses import dataclass, field
33
from typing import TYPE_CHECKING, Dict, Final, List, Optional, TypedDict
44

5-
from pydantic import root_validator
5+
from pydantic import model_validator
66
from pydantic.fields import Field
77

88
from datahub.configuration.common import AllowDenyPattern, ConfigModel, LaxStr
@@ -271,7 +271,8 @@ class KafkaConnectSourceConfig(
271271

272272
stateful_ingestion: Optional[StatefulStaleMetadataRemovalConfig] = None
273273

274-
@root_validator(pre=True)
274+
@model_validator(mode="before")
275+
@classmethod
275276
def auto_construct_connect_uri(cls, values: Dict) -> Dict:
276277
"""
277278
Auto-construct connect_uri from Confluent Cloud environment and cluster IDs.

0 commit comments

Comments
 (0)