From bf6071475028cf9711b8264e8952deb2aee2e32b Mon Sep 17 00:00:00 2001
From: Daniele Grattarola <daniele.grattarola@gmail.com>
Date: Sun, 21 Jan 2024 16:34:25 +0000
Subject: [PATCH] Update sklearn OHE

---
 spektral/datasets/tudataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spektral/datasets/tudataset.py b/spektral/datasets/tudataset.py
index 38775352..f26b0de8 100644
--- a/spektral/datasets/tudataset.py
+++ b/spektral/datasets/tudataset.py
@@ -216,7 +216,7 @@ def _normalize(x, norm=None):
     Apply one-hot encoding or z-score to a list of node features
     """
     if norm == "ohe":
-        fnorm = OneHotEncoder(sparse=False, categories="auto")
+        fnorm = OneHotEncoder(sparse_output=False, categories="auto")
     elif norm == "zscore":
         fnorm = StandardScaler()
     else: