diff --git a/flytekit/core/type_engine.py b/flytekit/core/type_engine.py index bcd34cf6f3..bdd8460c39 100644 --- a/flytekit/core/type_engine.py +++ b/flytekit/core/type_engine.py @@ -1532,7 +1532,6 @@ async def _literal_map_to_kwargs( raise TypeTransformerFailedError( f"Error converting input '{k}' at position {i}:\n" f"Literal value: {lm.literals[k]}\n" - f"Literal type: {literal_types}\n" f"Expected Python type: {python_interface_inputs[k]}\n" f"Exception: {e}" ) diff --git a/plugins/flytekit-greatexpectations/tests/test_schema.py b/plugins/flytekit-greatexpectations/tests/test_schema.py index f9a63634ad..a418672581 100644 --- a/plugins/flytekit-greatexpectations/tests/test_schema.py +++ b/plugins/flytekit-greatexpectations/tests/test_schema.py @@ -5,6 +5,8 @@ import pandas as pd import pytest + +from flytekit.core.type_engine import TypeTransformerFailedError from flytekitplugins.great_expectations import BatchRequestConfig, GreatExpectationsFlyteConfig, GreatExpectationsType from great_expectations.exceptions import InvalidBatchRequestError, ValidationError @@ -58,7 +60,7 @@ def invalid_wf(dataset: str = "yellow_tripdata_sample_2019-02.csv") -> int: valid_result = valid_wf() assert valid_result == 9999 - with pytest.raises(ValidationError): + with pytest.raises(TypeTransformerFailedError): invalid_wf() @@ -144,7 +146,7 @@ def my_wf(): my_task(directory="my_assets") # Capture IndexError - with pytest.raises(InvalidBatchRequestError): + with pytest.raises(TypeTransformerFailedError): my_wf() diff --git a/tests/flytekit/integration/remote/test_remote.py b/tests/flytekit/integration/remote/test_remote.py index 3babe86c45..8c6e55b5cf 100644 --- a/tests/flytekit/integration/remote/test_remote.py +++ b/tests/flytekit/integration/remote/test_remote.py @@ -917,7 +917,7 @@ def test_attr_access_sd(): remote = FlyteRemote(Config.auto(config_file=CONFIG), PROJECT, DOMAIN) execution = remote.fetch_execution(name=execution_id) execution = remote.wait(execution=execution, timeout=datetime.timedelta(minutes=5)) - print("Execution Error:", execution.error) + assert execution.error is None, f"Execution failed with error: {execution.error}" assert execution.closure.phase == WorkflowExecutionPhase.SUCCEEDED, f"Execution failed with phase: {execution.closure.phase}" # Delete the remote file to free the space