fix numerical features order issue

OHDSI · Oct 30, 2023 · 20e855b · 20e855b
1 parent dac66a0
commit 20e855b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inst/python/Dataset.py b/inst/python/Dataset.py
@@ -67,7 +67,7 @@ def __init__(self,
         if pl.count(self.numerical_features) == 0:
             self.num = None
         else:
-            numerical_data = data.filter(pl.col('columnId').is_in(self.numerical_features)). \
+            numerical_data = data.filter(pl.col('columnId').is_in(self.numerical_features)).sort(by='columnId'). \
                 with_row_count('newColumnId').with_columns(pl.col('newColumnId').first().over('columnId').
                                                            rank(method="dense") - 1, pl.col('rowId') - 1) \
                 .select(pl.col('rowId'), pl.col('newColumnId').alias('columnId'), pl.col('covariateValue')).collect()