Update _doc/examples/ml/plot_template_data.py

sdpython · Copilot · web-flow · commit 355cf1f6f42a · 2026-04-12T10:21:57.000+02:00
Co-authored-by: Copilot &lt;175728472+Copilot@users.noreply.github.com&gt;
diff --git a/_doc/examples/ml/plot_template_data.py b/_doc/examples/ml/plot_template_data.py
@@ -72,9 +72,14 @@ def compute_oracle(table, cible):
             columns="Session",
             values=cible,
         )
-        # .dropna(axis=0)  # fails
         .sort_index()
     )
+    # Keep only rows where both 2024 and 2025 have non-missing values
+    piv = piv.dropna(axis=0, how="any")
+    if piv.empty:
+        raise ValueError(
+            "Not enough overlapping data between 2024 and 2025 to compute oracle."
+        )
     return mean_absolute_error(piv[2025], piv[2024])
 
 

Original file line number	Diff line number	Diff line change
`@@ -72,9 +72,14 @@ def compute_oracle(table, cible):`
`72`	`72`	`columns="Session",`
`73`	`73`	`values=cible,`
`74`	`74`	`)`
`75`		`- # .dropna(axis=0) # fails`
`76`	`75`	`.sort_index()`
`77`	`76`	`)`
	`77`	`+ # Keep only rows where both 2024 and 2025 have non-missing values`
	`78`	`+ piv = piv.dropna(axis=0, how="any")`
	`79`	`+ if piv.empty:`
	`80`	`+ raise ValueError(`
	`81`	`+ "Not enough overlapping data between 2024 and 2025 to compute oracle."`
	`82`	`+ )`
`78`	`83`	`return mean_absolute_error(piv[2025], piv[2024])`
`79`	`84`
`80`	`85`