fix imports

Author: aaronsteers

airbyte_cdk/test/entrypoint_wrapper.py

@@ -38,14 +38,14 @@
     AirbyteMessageSerializer,
     AirbyteStateMessage,
     AirbyteStateMessageSerializer,
+    AirbyteStreamState,
     AirbyteStreamStatus,
     ConfiguredAirbyteCatalog,
     ConfiguredAirbyteCatalogSerializer,
     Level,
     TraceType,
     Type,
 )
-from airbyte_cdk.models.airbyte_protocol import AirbyteMessage, AirbyteStreamState
 from airbyte_cdk.sources import Source
 from airbyte_cdk.test.models.scenario import ExpectedOutcome