From 5c0c3285910b43fc5f09b5df71130d5ad938a9ca Mon Sep 17 00:00:00 2001 From: James Brandreth Date: Wed, 28 Jul 2021 15:17:17 +0100 Subject: [PATCH] :bug: leave as lists --- .../mixins/query_mixins/data_config_query_mixin.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/containers/cleanair/cleanair/mixins/query_mixins/data_config_query_mixin.py b/containers/cleanair/cleanair/mixins/query_mixins/data_config_query_mixin.py index 187f95713..ec7b16ff1 100644 --- a/containers/cleanair/cleanair/mixins/query_mixins/data_config_query_mixin.py +++ b/containers/cleanair/cleanair/mixins/query_mixins/data_config_query_mixin.py @@ -27,7 +27,6 @@ def query_data_config( Args: train_start_date: ISO formatted datetime of the start of the training period. pred_start_date: ISO formatted datetime of the start of the prediciton period. - features: list of feature names, searches for exact match, ignoring order Returns: A database query with columns for data id, data config and preprocessing. @@ -47,12 +46,12 @@ def query_data_config( ) if static_features: data_ids = data_ids.filter( - set(AirQualityDataTable.data_config["features"]) - == set(static_features) + AirQualityDataTable.data_config["features"] + == static_features ) if dynamic_features: data_ids = data_ids.filter( - set(AirQualityDataTable.data_config["features"]) - == set(dynamic_features) + AirQualityDataTable.data_config["features"] + == dynamic_features ) return data_ids