hotfix for new polars count (#107)

Fixes #106
OHDSI · Dec 22, 2023 · 6e4b523 · 6e4b523
1 parent f3340d9
commit 6e4b523
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/inst/python/Dataset.py b/inst/python/Dataset.py
@@ -69,7 +69,7 @@ def __init__(self, data, labels=None, numerical_features=None):
 
         # numerical data,
         # N x C, dense matrix with values for N patients/visits for C numerical features
-        if pl.count(self.numerical_features) == 0:
+        if self.numerical_features.count() == 0:
             self.num = None
         else:
             map_numerical = dict(zip(self.numerical_features.sort().to_list(),
@@ -96,7 +96,7 @@ def __init__(self, data, labels=None, numerical_features=None):
             self.num = torch.sparse_coo_tensor(
                 indices=indices.T,
                 values=values.squeeze(),
-                size=(observations, pl.count(self.numerical_features)),
+                size=(observations, self.numerical_features.count()),
             ).to_dense()
         delta = time.time() - start
         print(f"Processed data in {delta:.2f} seconds")