fix failed test

Author: ashleyxuu

tests/system/small/ml/test_llm.py

@@ -285,8 +285,8 @@ def test_embedding_generator_predict_series_success(
 ):
     df = palm2_embedding_generator_model.predict(llm_text_df["prompt"]).to_pandas()
     assert df.shape == (3, 4)
-    assert "text_embedding" in df.columns
-    series = df["text_embedding"]
+    assert "ml_generate_embedding_result" in df.columns
+    series = df["ml_generate_embedding_result"]
     value = series[0]
     assert len(value) == 768