refactor: derive the _schema automatically

opentargets · Dec 3, 2024 · 43a710f · 43a710f
1 parent a02f9c1
commit 43a710f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/gentropy/dataset/dataset.py b/src/gentropy/dataset/dataset.py
@@ -32,7 +32,6 @@ class Dataset(ABC):
     """
 
     _df: DataFrame
-    _schema: StructType
 
     def __post_init__(self: Dataset) -> None:
         """Post init."""
@@ -64,7 +63,7 @@ def schema(self: Dataset) -> StructType:
         Returns:
             StructType: Dataframe expected schema
         """
-        return self._schema
+        return self.get_schema()
 
     @classmethod
     def _process_class_params(
@@ -172,7 +171,7 @@ def validate_schema(self: Dataset) -> None:
         Raises:
             SchemaValidationError: If the DataFrame schema does not match the expected schema
         """
-        expected_schema = self._schema
+        expected_schema = self.schema
         observed_schema = self._df.schema
 
         # Unexpected fields in dataset