From b922a4eb4d8828ef37f0e4796ec64381e3ef6f87 Mon Sep 17 00:00:00 2001
From: Ines Oliveira e Silva <inesoliveiraesilva@gmail.com>
Date: Thu, 29 Feb 2024 10:24:18 +0000
Subject: [PATCH] Fix load data function (#180)

---
 src/aequitas/flow/datasets/generic.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/aequitas/flow/datasets/generic.py b/src/aequitas/flow/datasets/generic.py
index e8c578f8..02b336de 100644
--- a/src/aequitas/flow/datasets/generic.py
+++ b/src/aequitas/flow/datasets/generic.py
@@ -152,7 +152,7 @@ def _validate_splits(self) -> None:
     def load_data(self) -> None:
         """Load the dataset."""
         self.logger.info("Loading data.")
-        if hasattr(self, "data"):
+        if self._data is not None:
             return
         if self.extension == "parquet":
             read_method = pd.read_parquet