fix(ingest/dremio): fix report, mark usage stats capability (#13851)

This commit is contained in:
Aseem Bansal 2025-06-25 15:09:52 +05:30 committed by GitHub
parent 3b44ed847c
commit a9e9ac9808
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -89,6 +89,7 @@ class DremioSourceMapEntry:
@capability(SourceCapability.LINEAGE_COARSE, "Enabled by default")
@capability(SourceCapability.OWNERSHIP, "Enabled by default")
@capability(SourceCapability.PLATFORM_INSTANCE, "Enabled by default")
@capability(SourceCapability.USAGE_STATS, "Enabled by default to get usage stats")
class DremioSource(StatefulIngestionSourceBase):
"""
This plugin integrates with Dremio to extract and ingest metadata into DataHub.
@ -242,7 +243,6 @@ class DremioSource(StatefulIngestionSourceBase):
# Generate workunit for aggregated SQL parsing results
for mcp in self.sql_parsing_aggregator.gen_metadata():
self.report.report_workunit(mcp.as_workunit())
yield mcp.as_workunit()
# Profiling