Skip to content

[BUG] fix MapieRegressor dependency check, and BaseProbaRegressor._predict_proba default#540

Merged
fkiraly merged 3 commits into
mainfrom
fix-mapieregressor
May 8, 2025
Merged

[BUG] fix MapieRegressor dependency check, and BaseProbaRegressor._predict_proba default#540
fkiraly merged 3 commits into
mainfrom
fix-mapieregressor

Conversation

@fkiraly
Copy link
Copy Markdown
Collaborator

@fkiraly fkiraly commented May 6, 2025

This PR fixes two unreported bugs:

  • the MapieRegressor dependency check was always failing due to the capitalization of MAPIE on pypi - not mapie. This led to the estimator not being useable since the upgrade to strict PEP 440 (which included capitalization sensitivity) of the python_dependencies tag.
  • the BaseProbaRegressor._predict_proba sensible default would not work properly if the expected return of predict was not a pd.DataFrame. This has been remedied by using the idiomatic method _get_columns to determine the column names.

@fkiraly fkiraly added bug module:regression probabilistic regression module module:base-framework BaseObject, registry, base framework labels May 6, 2025
@fkiraly fkiraly merged commit f0f51d8 into main May 8, 2025
34 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug module:base-framework BaseObject, registry, base framework module:regression probabilistic regression module

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant