Closes OPEN-4602 Fix issue with pandas Series for text classification commit bundle validation

gustavocidornelas · whoseoyster · commit 43847fd40319 · 2023-07-13T11:03:54.000-07:00
diff --git a/openlayer/validators/commit_validators.py b/openlayer/validators/commit_validators.py
@@ -391,7 +391,7 @@ def _get_sample_input_data(self) -> Optional[pd.DataFrame]:
         )
         if validation_dataset_df is not None:
             sample_data = validation_dataset_df[
-                self.validation_dataset_config["textColumnName"]
+                [self.validation_dataset_config["textColumnName"]]
             ].head()
 
         return sample_data

Original file line number	Diff line number	Diff line change
`@@ -391,7 +391,7 @@ def _get_sample_input_data(self) -> Optional[pd.DataFrame]:`
`391`	`391`	`)`
`392`	`392`	`if validation_dataset_df is not None:`
`393`	`393`	`sample_data = validation_dataset_df[`
`394`		`- self.validation_dataset_config["textColumnName"]`
	`394`	`+ [self.validation_dataset_config["textColumnName"]]`
`395`	`395`	`].head()`
`396`	`396`
`397`	`397`	`return sample_data`