Merge pull request #139 from scikit-learn-contrib/dev

doc examples patched
scikit-learn-contrib · Apr 17, 2024 · 41884b2 · 41884b2
2 parents 3fba238 + 4e5a380
commit 41884b2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/tutorials/plot_tuto_categorical.py b/examples/tutorials/plot_tuto_categorical.py
@@ -21,7 +21,7 @@
 # ---------------------------------------------------------------
 # We get the data and focus on the explanatory variables
 df = data.get_data("Titanic")
-df = df.drop(columns=["survived"])
+df = df.drop(columns=["Survived"])
 
 # %%
 # 2. Mixed type imputation methods

diff --git a/qolmat/utils/data.py b/qolmat/utils/data.py
@@ -324,7 +324,7 @@ def add_holes(df: pd.DataFrame, ratio_masked: float, mean_size: int) -> pd.DataF
     pd.DataFrame
         dataframe with missing values
     """
-    groups = df.index.names.difference(["datetime", "date", "index"])
+    groups = df.index.names.difference(["datetime", "date", "index", None])
     if groups != []:
         generator = missing_patterns.GeometricHoleGenerator(
             1, ratio_masked=ratio_masked, subset=df.columns, groups=groups