Skip to content

Conversation

@flying-sheep
Copy link
Collaborator

@flying-sheep flying-sheep commented Dec 17, 2024

To use the new API, we need to depend on sklearn 1.6. But since we don’t really change much, the resolver will resolve our package to an older version if people want to use sklearn<1.6, so it should be fine.

I changed the way transformer check failures are handled in the same way as they did in scikit-learn/scikit-learn#30149

return Tags(
estimator_type="transformer",
target_tags=TargetTags(required=False),
transformer_tags=TransformerTags(),
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The default for preserves_dtype is ["float64"], is that accurate here?

@flying-sheep
Copy link
Collaborator Author

@frankier could you please take a look?

@flying-sheep flying-sheep merged commit eaae17c into main Jan 13, 2025
4 checks passed
@flying-sheep flying-sheep deleted the pa/sklearn-1.6 branch January 13, 2025 15:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants