clean up imports

Author: aaronsteers

airbyte_cdk/test/standard_tests/_job_runner.py

@@ -18,14 +18,14 @@
     ConnectorSpecification,
     Status,
 )
-from airbyte_cdk.test import entrypoint_wrapper
+from airbyte_cdk.test.entrypoint_wrapper import EntrypointOutput, _run_command
 from airbyte_cdk.test.standard_tests.models import (
     ConnectorTestScenario,
 )
 
 
 def _errors_to_str(
-    entrypoint_output: entrypoint_wrapper.EntrypointOutput,
+    entrypoint_output: EntrypointOutput,
 ) -> str:
     """Convert errors from entrypoint output to a string."""
     if not entrypoint_output.errors:
@@ -65,7 +65,7 @@ def run_test_job(
     *,
     test_scenario: ConnectorTestScenario | None = None,
     catalog: ConfiguredAirbyteCatalog | dict[str, Any] | None = None,
-) -> entrypoint_wrapper.EntrypointOutput:
+) -> EntrypointOutput:
     """Run a test scenario from provided CLI args and return the result."""
     # Use default (empty) scenario if not provided:
     test_scenario = test_scenario or ConnectorTestScenario()
@@ -120,7 +120,7 @@ def run_test_job(
     # This is a bit of a hack because the source needs the catalog early.
     # Because it *also* can fail, we have to redundantly wrap it in a try/except block.
 
-    result: entrypoint_wrapper.EntrypointOutput = entrypoint_wrapper._run_command(  # noqa: SLF001  # Non-public API
+    result: EntrypointOutput = _run_command(  # noqa: SLF001  # Non-public API
         source=connector_obj,  # type: ignore [arg-type]
         args=args,
         expecting_exception=test_scenario.expect_exception,