Merge 734419f into 9137e86

skit-ai · Sep 7, 2021 · 3acc504 · 3acc504
2 parents 9137e86 + 734419f
commit 3acc504
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dialogy/plugins/text/merge_asr_output/__init__.py b/dialogy/plugins/text/merge_asr_output/__init__.py
@@ -106,10 +106,10 @@ def transform(self, training_data: pd.DataFrame) -> pd.DataFrame:
         training_data["use"] = True
         logger.debug("Transforming training data.")
         for i, row in training_data.iterrows():
+            asr_output = None
             try:
                 asr_output = json.loads(row[self.data_column])
-                if asr_output:
-                    merged_asr_ouptut = merge_asr_output(asr_output)
+                if asr_output and (merged_asr_ouptut := merge_asr_output(asr_output)):
                     training_data.loc[i, self.data_column] = merged_asr_ouptut[0]
                 else:
                     training_data.loc[i, "use"] = False