formatting

Author: andreatgretel

examples/custom_column.py

@@ -42,7 +42,7 @@
             "finance",
             "technology",
         ]
-    }
+    },
 )
 
 builder.add_column(
@@ -53,10 +53,12 @@
     system_prompt=SYSTEM_PROMPT,
 )
 
+
 def generator_function(df: pd.DataFrame) -> pd.DataFrame:
     df["length_frac"] = df["text"].apply(lambda x: len(x) / 1000)
     return df
 
+
 builder.add_column(
     CustomColumnConfig(
         name="length_frac",
@@ -69,4 +71,4 @@ def generator_function(df: pd.DataFrame) -> pd.DataFrame:
 dd_preview = dd.preview(builder, num_records=10)
 dd_preview.display_sample_record()
 
-dd.create(builder, num_records=20)
+dd.create(builder, num_records=20)