minor bug fix self not included in some function

Author: Teris45

pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pyjedai"
-version = "0.2.8"
+version = "0.2.9"
 description = "An open-source library that builds powerful end-to-end Entity Resolution workflows."
 readme = "README.md"
 authors = [

src/pyjedai/schema/schema_model.py

@@ -29,8 +29,7 @@ def __init__(
 
         if not isinstance(dataset_2, pd.DataFrame):
             raise AttributeError("Dataset 2 must be a pandas DataFrame")
-            
-        
+                    
         if matching_type == 'CONTENT':
             dataset_1, dataset_2, ground_truth = self.load_content(dataset_1, dataset_2, ground_truth, skip_ground_truth_processing)
         elif matching_type == 'COMPOSITE': 
@@ -55,7 +54,6 @@ def load_content(self,
                 ground_truth: DataFrame = None,
                 skip_ground_truth_processing: bool = False) -> tuple:
 
-
         dataset_1 = dataset_1.astype(str)
         dataset_2 = dataset_2.astype(str)
 
@@ -102,8 +100,7 @@ def load_content(self,
         self.ground_truth = ground_truth      
         return dataset_1, dataset_2, ground_truth
 
-
-    def load_composite(
+    def load_composite(self,
             dataset_1: DataFrame, 
             dataset_2: DataFrame,
             ground_truth: DataFrame = None,
@@ -155,13 +152,13 @@ def load_composite(
         return dataset_1, dataset_2, ground_truth
 
 
-
-    def load_schema(
+    def load_schema(self,
             dataset_1: DataFrame, 
             dataset_2: DataFrame,
             ground_truth: DataFrame = None,
             skip_ground_truth_processing: bool = False
     ) -> tuple:
+        
         dataset_1 = dataset_1.astype(str)
         dataset_2 = dataset_2.astype(str)