diff --git a/mipengine/algorithms/multiple_histograms.py b/mipengine/algorithms/multiple_histograms.py index 3efda5185611787a9ba06a2b897955e556bee3c9..4c010621368172138cedd6325ea28b351bc943a7 100644 --- a/mipengine/algorithms/multiple_histograms.py +++ b/mipengine/algorithms/multiple_histograms.py @@ -41,6 +41,7 @@ class HistogramAlgorithm(Algorithm, algname="multiple_histograms"): xvars = self.executor.x_variables or [] yvars = self.executor.y_variables or [] + yvar = yvars[0] default_bins = 20 bins = self.executor.algorithm_parameters.get("bins", default_bins) @@ -51,13 +52,12 @@ class HistogramAlgorithm(Algorithm, algname="multiple_histograms"): metadata = dict(self.executor.metadata) - vars = [var for var in xvars + yvars if var != "dataset"] + vars = [var for var in xvars + yvars] nominal_vars = [var for var in vars if metadata[var]["is_categorical"]] enumerations_dict = {var: metadata[var]["enumerations"] for var in nominal_vars} - yvar = yvars[0] if yvar in xvars: xvars.remove(yvar)