test: skip checking llm tuning score labels (#1340)

Author: GarrettWu

tests/system/small/ml/test_llm.py

@@ -868,7 +868,6 @@ def test_llm_palm_score_params(llm_fine_tune_df_default_index):
             "label",
             "evaluation_status",
         ],
-        index=6,
     )
 
 
@@ -928,7 +927,6 @@ def test_llm_gemini_pro_score_params(llm_fine_tune_df_default_index, model_name)
             "label",
             "evaluation_status",
         ],
-        index=6,
     )
 
 

tests/system/utils.py

@@ -298,23 +298,25 @@ def assert_pandas_df_equal_pca(actual, expected, **kwargs):
 def check_pandas_df_schema_and_index(
     pd_df: pd.DataFrame,
     columns: Iterable,
-    index: Union[int, Iterable],
+    index: Optional[Union[int, Iterable]] = None,
     col_exact: bool = True,
 ):
     """Check pandas df schema and index. But not the values.
 
     Args:
         pd_df: the input pandas df
         columns: target columns to check with
-        index: int or Iterable. If int, only check the length (index size) of the df. If Iterable, check index values match
+        index: int or Iterable or None, default None. If int, only check the length (index size) of the df. If Iterable, check index values match. If None, skip checking index.
         col_exact: If True, check the columns param are exact match. Otherwise only check the df contains all of those columns
     """
     if col_exact:
         assert list(pd_df.columns) == list(columns)
     else:
         assert set(columns) <= set(pd_df.columns)
 
-    if isinstance(index, int):
+    if index is None:
+        pass
+    elif isinstance(index, int):
         assert len(pd_df) == index
     elif isinstance(index, Iterable):
         assert list(pd_df.index) == list(index)