full upload so not to lose anything important

Deep-SAD-PyTorch/src/datasets/main.py

@@ -18,6 +18,9 @@ def load_dataset(
     ratio_pollution: float = 0.0,
     random_state=None,
     inference: bool = False,
+    k_fold: bool = False,
+    num_known_normal: int = 0,
+    num_known_outlier: int = 0,
 ):
     """Loads the dataset."""
 
@@ -46,6 +49,9 @@ def load_dataset(
             ratio_known_outlier=ratio_known_outlier,
             ratio_pollution=ratio_pollution,
             inference=inference,
+            k_fold=k_fold,
+            num_known_normal=num_known_normal,
+            num_known_outlier=num_known_outlier,
         )
 
     if dataset_name == "subtersplit":