Fix test_invalid_ge_schema_batchrequest_pandas_config (#3145)

Signed-off-by: Kevin Su <[email protected]>

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}"
             )

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()
 
 

tests/flytekit/integration/remote/test_remote.py

@@ -966,7 +966,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