fix name

Arize-ai · Nov 2, 2023 · ea0b642 · ea0b642
1 parent 4f38540
commit ea0b642
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/src/phoenix/trace/trace_eval_dataset.py b/src/phoenix/trace/trace_eval_dataset.py
@@ -74,7 +74,7 @@ def get_eval_dataframe(self, prefix_columns_with_name: bool = True) -> pd.DataFr
         return dataframe
 
 
-def binary_classifications_to_trace_evals_dataset(
+def binary_classifications_to_trace_eval_dataset(
     eval_name: str,
     classifications_df: pd.DataFrame,
     spans_df: pd.DataFrame,

diff --git a/tests/trace/test_trace_eval_dataset.py b/tests/trace/test_trace_eval_dataset.py
@@ -3,19 +3,19 @@
 import pandas as pd
 from phoenix.trace.trace_eval_dataset import (
     TraceEvalDataset,
-    binary_classifications_to_trace_evals_dataset,
+    binary_classifications_to_trace_eval_dataset,
 )
 
 
-def test_binary_classifications_to_trace_evals_dataset():
+def test_binary_classifications_to_trace_eval_dataset():
     # Create mock data
     eval_name = "toxicity"
     classifications_df = pd.DataFrame({"label": ["toxic", "non-toxic", "toxic", "non-toxic"]})
     spans_df = pd.DataFrame({"span_id": [1, 2, 3, 4], "other_column": ["a", "b", "c", "d"]})
     rails_map = OrderedDict({True: "toxic", False: "non-toxic"})
 
     # Run the function
-    result = binary_classifications_to_trace_evals_dataset(
+    result = binary_classifications_to_trace_eval_dataset(
         eval_name, classifications_df, spans_df, rails_map
     )
 

diff --git a/tutorials/internal/trace_evals.ipynb b/tutorials/internal/trace_evals.ipynb
@@ -96,9 +96,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from phoenix.trace.trace_eval_dataset import binary_classifications_to_trace_evals_dataset\n",
+    "from phoenix.trace.trace_eval_dataset import binary_classifications_to_trace_eval_dataset\n",
     "\n",
-    "evaluation_ds = binary_classifications_to_trace_evals_dataset(\n",
+    "evaluation_ds = binary_classifications_to_trace_eval_dataset(\n",
     "     eval_name=\"toxicity\",\n",
     "     classifications_df=eval_classifications_df,\n",
     "     spans_df=spans_df,\n",