Update integ test to wait for lambda call

Author: mollyheamazon

sagemaker-train/tests/integ/ai_registry/test_evaluator.py

@@ -103,6 +103,11 @@ def test_create_reward_function_from_local_py_file_and_invoke(
         assert evaluator.method == EvaluatorMethod.BYOC
         assert evaluator.reference is not None
 
+        # Wait for Lambda to become Active before invoking
+        lambda_client = boto3.client("lambda")
+        waiter = lambda_client.get_waiter("function_active_v2")
+        waiter.wait(FunctionName=evaluator.reference)
+
         # Invoke the Lambda directly to verify the handler is correct
         lambda_client = boto3.client("lambda")
         response = lambda_client.invoke(