From bec6e300f744dfdd3aa4b4629eaddc1c71a44b50 Mon Sep 17 00:00:00 2001 From: Andrew Sikowitz Date: Mon, 3 Feb 2025 14:35:15 -0800 Subject: [PATCH] lint --- .../src/datahub/ingestion/graph/entity_versioning.py | 6 ++---- smoke-test/tests/entity_versioning/test_versioning.py | 2 +- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/metadata-ingestion/src/datahub/ingestion/graph/entity_versioning.py b/metadata-ingestion/src/datahub/ingestion/graph/entity_versioning.py index 6d9f44ed7ef0c9..3d2288fdc627ab 100644 --- a/metadata-ingestion/src/datahub/ingestion/graph/entity_versioning.py +++ b/metadata-ingestion/src/datahub/ingestion/graph/entity_versioning.py @@ -17,16 +17,14 @@ def execute_graphql( variables: Optional[Dict], operation_name: Optional[str] = None, format_exception: bool = True, - ) -> Dict: - ... + ) -> Dict: ... def get_aspect( self, entity_urn: str, aspect_type: Type[Aspect], version: int = 0, - ) -> Optional[Aspect]: - ... + ) -> Optional[Aspect]: ... class EntityVersioningAPI(DataHubGraphProtocol): diff --git a/smoke-test/tests/entity_versioning/test_versioning.py b/smoke-test/tests/entity_versioning/test_versioning.py index 1c6dc3279cb3ba..c6e25b3418e550 100644 --- a/smoke-test/tests/entity_versioning/test_versioning.py +++ b/smoke-test/tests/entity_versioning/test_versioning.py @@ -1,4 +1,5 @@ import pytest + from datahub.emitter.mcp import MetadataChangeProposalWrapper from datahub.ingestion.graph.client import DataHubGraph from datahub.metadata.schema_classes import ( @@ -7,7 +8,6 @@ VersionSetPropertiesClass, ) from datahub.metadata.urns import DatasetUrn, VersionSetUrn - from tests.consistency_utils import wait_for_writes_to_sync VERSION_SET_URN = VersionSetUrn("12345678910", DatasetUrn.ENTITY_TYPE).urn()