Fix other partition routers

airbytehq · May 16, 2024 · 11851ee · 11851ee
1 parent 5172a32
commit 11851ee
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/...yte-cdk/python/airbyte_cdk/sources/declarative/partition_routers/list_partition_router.py b/...yte-cdk/python/airbyte_cdk/sources/declarative/partition_routers/list_partition_router.py
@@ -7,12 +7,12 @@
 
 from airbyte_cdk.sources.declarative.interpolation.interpolated_string import InterpolatedString
 from airbyte_cdk.sources.declarative.requesters.request_option import RequestOption, RequestOptionType
-from airbyte_cdk.sources.declarative.stream_slicers.stream_slicer import StreamSlicer
+from airbyte_cdk.sources.declarative.partition_routers.partition_router import PartitionRouter
 from airbyte_cdk.sources.types import Config, StreamSlice, StreamState
 
 
 @dataclass
-class ListPartitionRouter(StreamSlicer):
+class ListPartitionRouter(PartitionRouter):
     """
     Partition router that iterates over the values of a list
     If values is a string, then evaluate it as literal and assert the resulting literal is a list

diff --git a/...e-cdk/python/airbyte_cdk/sources/declarative/partition_routers/single_partition_router.py b/...e-cdk/python/airbyte_cdk/sources/declarative/partition_routers/single_partition_router.py
@@ -5,12 +5,12 @@
 from dataclasses import InitVar, dataclass
 from typing import Any, Iterable, Mapping, Optional
 
-from airbyte_cdk.sources.declarative.stream_slicers.stream_slicer import StreamSlicer
+from airbyte_cdk.sources.declarative.partition_routers.partition_router import PartitionRouter
 from airbyte_cdk.sources.types import StreamSlice, StreamState
 
 
 @dataclass
-class SinglePartitionRouter(StreamSlicer):
+class SinglePartitionRouter(PartitionRouter):
     """Partition router returning only a stream slice"""
 
     parameters: InitVar[Mapping[str, Any]]