fix: add missing colon

DataChefHQ · Sep 27, 2024 · 2d7fae9 · 2d7fae9
1 parent 2de6e44
commit 2d7fae9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sparkle/writer/kafka_writer.py b/src/sparkle/writer/kafka_writer.py
@@ -183,7 +183,7 @@ def write(self, df: DataFrame) -> None:
         # Convert the DataFrame to a Kafka-friendly format
         kafka_df = to_kafka_dataframe(self.unique_identifier_column_name, df)
 
-        if set(kafka_df.columns) != {"key", "value"}
+        if set(kafka_df.columns) != {"key", "value"}:
             raise KeyError(
                 "The DataFrame must contain 'key' and 'value' columns. "
                 "Ensure that `to_kafka_dataframe` transformation is correctly applied."