use passed dataframe

Author: chrishalcrow

src/spikeinterface/curation/model_based_curation.py

@@ -188,14 +188,9 @@ def _check_params_for_classification(self, enforce_metric_params=False, model_in
                     else:
                         warnings.warn(warning_message)
 
-    def _export_to_phy(self, classified_units):
+    def _export_to_phy(self, classified_df):
         """Export the classified units to Phy as cluster_prediction.tsv file"""
 
-        import pandas as pd
-
-        # Create a new DataFrame with unit_id, prediction, and probability columns from dict {unit_id: (prediction, probability)}
-        classified_df = pd.DataFrame.from_dict(classified_units, orient="index", columns=["prediction", "probability"])
-
         # Export to Phy format
         try:
             sorting_path = self.sorting_analyzer.sorting.get_annotation("phy_folder")