adding small features to cost model

mitdbg · Nov 2, 2023 · 04949cb · 04949cb
1 parent 2560ca2
commit 04949cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brad/cost_model/dataset/dataset_creation.py b/src/brad/cost_model/dataset/dataset_creation.py
@@ -405,7 +405,7 @@ def create_query_dataloader(
             query_collator,
             db_statistics=val_database_statistics,
             feature_statistics=feature_statistics,
-            plan_featurization_name=query_featurization_name,
+            query_featurization_name=query_featurization_name,
         )
         dataloader_args = dict(
             batch_size=batch_size,