🐛 leave as lists

alan-turing-institute · Jul 28, 2021 · 010c4f7 · 010c4f7
1 parent a315052
commit 010c4f7
Showing 1 changed file with 4 additions and 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
@@ -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