diff --git a/deepeval/dataset/dataset.py b/deepeval/dataset/dataset.py index 939d5617..1aad97f8 100644 --- a/deepeval/dataset/dataset.py +++ b/deepeval/dataset/dataset.py @@ -420,7 +420,11 @@ def get_column_data(df: pd.DataFrame, col_name: str, default=None): else [default] * len(df) ) - df = pd.read_csv(file_path).astype(object).where(pd.notna(pd.read_csv(file_path)), None) + df = ( + pd.read_csv(file_path) + .astype(object) + .where(pd.notna(pd.read_csv(file_path)), None) + ) inputs = get_column_data(df, input_col_name) actual_outputs = get_column_data(df, actual_output_col_name) diff --git a/docs/docs/synthesizer-generate-from-contexts.mdx b/docs/docs/synthesizer-generate-from-contexts.mdx index 755444d9..e02aa31d 100644 --- a/docs/docs/synthesizer-generate-from-contexts.mdx +++ b/docs/docs/synthesizer-generate-from-contexts.mdx @@ -46,7 +46,7 @@ synthesizer.generate_goldens_from_contexts( ) ``` -There are one mandatory and two optional parameters when using the `generate_goldens_from_contexts` method: +There are one mandatory and three optional parameters when using the `generate_goldens_from_contexts` method: - `contexts`: a list of context, where each context is itself a list of strings, ideally sharing a common theme or subject area. - [Optional] `include_expected_output`: a boolean which when set to `True`, will additionally generate an `expected_output` for each synthetic `Golden`. Defaulted to `True`.