[JTH] fix bug in 2 vars plots in LHS

tausiaj · tausiaj · commit 3eceb48f2bf9 · 2025-04-16T16:40:36.000+02:00
diff --git a/bluemath_tk/datamining/_base_datamining.py b/bluemath_tk/datamining/_base_datamining.py
@@ -90,6 +90,8 @@ def plot_generated_data(
             sharex=False,
             sharey=False,
         )
+        if isinstance(axes, Axes):
+            axes = np.array([[axes]])
 
         for c1, v1 in enumerate(variables_names[1:]):
             for c2, v2 in enumerate(variables_names[:-1]):

Original file line number	Diff line number	Diff line change
`@@ -90,6 +90,8 @@ def plot_generated_data(`
`90`	`90`	`sharex=False,`
`91`	`91`	`sharey=False,`
`92`	`92`	`)`
	`93`	`+ if isinstance(axes, Axes):`
	`94`	`+ axes = np.array([[axes]])`
`93`	`95`
`94`	`96`	`for c1, v1 in enumerate(variables_names[1:]):`
`95`	`97`	`for c2, v2 in enumerate(variables_names[:-1]):`