From 368182deb9f437b3696ad93efee49dc8a55de164 Mon Sep 17 00:00:00 2001 From: Harshal Sheth Date: Fri, 9 Sep 2022 17:23:25 -0700 Subject: [PATCH] add extra dataclass check --- metadata-ingestion/src/datahub/ingestion/sink/sink_registry.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/metadata-ingestion/src/datahub/ingestion/sink/sink_registry.py b/metadata-ingestion/src/datahub/ingestion/sink/sink_registry.py index 97e3c9268e9bc2..3c364dc9249851 100644 --- a/metadata-ingestion/src/datahub/ingestion/sink/sink_registry.py +++ b/metadata-ingestion/src/datahub/ingestion/sink/sink_registry.py @@ -1,3 +1,4 @@ +import dataclasses from typing import Type from datahub.ingestion.api.registry import PluginRegistry @@ -5,6 +6,7 @@ def _check_sink_classes(cls: Type[Sink]) -> None: + assert not dataclasses.is_dataclass(cls), f"Sink {cls} is a dataclass" assert cls.get_config_class() assert cls.get_report_class()