save

databricks · Jan 22, 2024 · 8d5ccb6 · 8d5ccb6
1 parent ee67d3a
commit 8d5ccb6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lilac/data/clustering.py b/lilac/data/clustering.py
@@ -453,7 +453,8 @@ def compute_category_titles(items: Iterator[Item]) -> Iterator[Item]:
 
   def drop_temp_text_column(items: Iterator[Item]) -> Iterator[Item]:
     for item in items:
-      del item[TEXT_COLUMN]
+      if TEXT_COLUMN in item:
+        del item[TEXT_COLUMN]
       yield item
 
   # Drop the temporary newline-concatenated text column and write the final output.