Skip to content

Commit 9a47250

Browse files
Satish615Satish Gollaprolu
andauthored
fix: replace sklearn with scikit-learn for the python package (#134)
* fix: replace sklearn with scikit-learn for the python package Co-authored-by: Satish Gollaprolu <[email protected]>
1 parent 873e760 commit 9a47250

File tree

4 files changed

+8
-8
lines changed

4 files changed

+8
-8
lines changed

examples/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
sklearn
1+
scikit-learn
22
catboost
33
seaborn
44
pandas

setup.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
import os
55
from setuptools import find_packages, setup
66

7-
INSTALL_REQUIRES = ["boto3", "pyarrow", "pandas", "s3fs", "numpy", "sklearn", "pyfunctional"]
7+
INSTALL_REQUIRES = ["boto3", "pyarrow", "pandas", "s3fs", "numpy", "scikit-learn", "pyfunctional"]
88

99
EXTRAS_REQUIRE = {
1010
"test": [
@@ -33,7 +33,7 @@
3333

3434
setup(
3535
name="smclarify",
36-
version="0.2",
36+
version="0.3",
3737
packages=find_packages("src"),
3838
package_dir={"": "src"},
3939
package_data={"smclarify": data_files},

src/smclarify/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = "0.2"
1+
__version__ = "0.3"

src/smclarify/bias/report.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ def __init__(self, name: str, description: str, value: Optional[float]):
8383
class MetricError(MetricResult):
8484
"""Metric Result with name, description and computed metric value and error"""
8585

86-
def __init__(self, name: str, description: str, value: Optional[float] = None, error: Exception = None):
86+
def __init__(self, name: str, description: str, value: Optional[float] = None, error: Exception = None): # type: ignore
8787
super().__init__(name, description, value)
8888
self.error = str(error)
8989

@@ -359,7 +359,7 @@ def bias_report(
359359
facet_column: FacetColumn,
360360
label_column: LabelColumn,
361361
stage_type: StageType,
362-
predicted_label_column: LabelColumn = None,
362+
predicted_label_column: Optional[LabelColumn] = None,
363363
metrics: List[Any] = ["all"],
364364
group_variable: Optional[pd.Series] = None,
365365
) -> List[Dict]:
@@ -406,7 +406,7 @@ def bias_basic_stats(
406406
facet_column: FacetColumn,
407407
label_column: LabelColumn,
408408
stage_type: StageType,
409-
predicted_label_column: LabelColumn = None,
409+
predicted_label_column: Optional[LabelColumn] = None,
410410
) -> List[Dict]:
411411
"""Computes size and confusion matrix.
412412
@@ -431,7 +431,7 @@ def _report(
431431
label_column: LabelColumn,
432432
stage_type: StageType,
433433
methods: List[Callable],
434-
predicted_label_column: LabelColumn = None,
434+
predicted_label_column: Optional[LabelColumn] = None,
435435
group_variable: Optional[pd.Series] = None,
436436
) -> List[Dict]:
437437
"""

0 commit comments

Comments
 (0)