diff --git a/arroyo/processing/strategies/decoder/__init__.py b/arroyo/processing/strategies/decoder/__init__.py index 431a1af0..aa04714b 100644 --- a/arroyo/processing/strategies/decoder/__init__.py +++ b/arroyo/processing/strategies/decoder/__init__.py @@ -1,19 +1,17 @@ -from arroyo.processing.strategies.decoder.avro import AvroCodec +from arroyo.processing.strategies.decoder import avro, json, msgpack from arroyo.processing.strategies.decoder.base import ( Codec, DecodedKafkaMessage, KafkaMessageDecoder, ValidationError, ) -from arroyo.processing.strategies.decoder.json import JsonCodec -from arroyo.processing.strategies.decoder.msgpack import MsgpackCodec __all__ = [ - "AvroCodec", + "avro", + "json", + "msgpack", "Codec", "DecodedKafkaMessage", - "JsonCodec", "KafkaMessageDecoder", - "MsgpackCodec", "ValidationError", ] diff --git a/tests/processing/strategies/test_decoder.py b/tests/processing/strategies/test_decoder.py index e7dacaf8..3219598d 100644 --- a/tests/processing/strategies/test_decoder.py +++ b/tests/processing/strategies/test_decoder.py @@ -11,13 +11,13 @@ from arroyo.backends.kafka import KafkaPayload from arroyo.processing.strategies.decoder import ( - AvroCodec, DecodedKafkaMessage, - JsonCodec, KafkaMessageDecoder, - MsgpackCodec, ValidationError, ) +from arroyo.processing.strategies.decoder.avro import AvroCodec +from arroyo.processing.strategies.decoder.json import JsonCodec +from arroyo.processing.strategies.decoder.msgpack import MsgpackCodec from arroyo.types import Message, Partition, Topic, Value