From 04949cb20b12d111d6635186c602f26180f5d24e Mon Sep 17 00:00:00 2001 From: wuziniu Date: Thu, 2 Nov 2023 16:25:32 -0400 Subject: [PATCH] adding small features to cost model --- src/brad/cost_model/dataset/dataset_creation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/brad/cost_model/dataset/dataset_creation.py b/src/brad/cost_model/dataset/dataset_creation.py index 09c8990f..e5b54f20 100644 --- 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,