fix count for new polars

inst/python/Dataset.py

@@ -67,7 +67,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(
@@ -100,7 +100,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")
@@ -128,3 +128,8 @@ def __getitem__(self, item):
         ):
             batch["num"] = batch["num"].unsqueeze(0)
         return [batch, self.target[item].squeeze()]
+
+
+
+
+