search

aws / amazon-sagemaker-clarify

Fairness Aware Machine Learning. Bias detection and mitigation for datasets and models.
Apache License 2.0
67 stars 40 forks source link

fix: replace sklearn with scikit-learn for the python package #134

Closed Satish615 closed 1 year ago

Satish615 commented 1 year ago

Issue #, if available: The latest smclarify install utilizes a deprecated sklearn package installation. This sklearn package now triggers an error when installing sklearn over scikit-learn.

Description of changes:

  1. Replace the sklearn deprecated with scikit-learn package to fix the above issue.
  2. use Optional for LabelColumn type where default value is None to avoid mypy errors

Testing: pytest tests/integration/test_bias_metrics.py

============================================================================================ 1 passed, 2 warnings in 3.19s =============================================================================================

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.