Skip to content

Commit

Permalink
adding super().__init__() to all classes that inherit from BaseEstimator
Browse files Browse the repository at this point in the history
  • Loading branch information
brifordwylie committed Dec 9, 2024
1 parent 9d0bfe4 commit 5d0cb2f
Show file tree
Hide file tree
Showing 6 changed files with 6 additions and 3 deletions.
2 changes: 1 addition & 1 deletion applications/aws_dashboard/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ sagemaker >= 2.143
cryptography>=42.0.5
ipython>=8.17.2
xgboost>=2.0.3
scikit-learn >=1.4.2, <=1.52
scikit-learn >=1.4.2
joblib>=1.3.2
requests>=2.32.0
plotly >= 5.18.0
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ dependencies = [
"ipython >= 8.17.2",
"pyreadline3; sys_platform == 'win32'",
"xgboost >= 2.0.3",
"scikit-learn >=1.4.2, <=1.52",
"scikit-learn >=1.4.2",
"joblib >= 1.3.2",
"requests >= 2.26.0"
]
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,6 @@ cryptography>=42.0.5
ipython>=8.17.2
pyreadline3; sys_platform == 'win32'
xgboost>=2.0.3
scikit-learn >=1.4.2, <=1.52
scikit-learn >=1.4.2
joblib>=1.3.2
requests>=2.32.0
1 change: 1 addition & 0 deletions src/sageworks/algorithms/dataframe/quantile_regression.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ def __init__(
self.q_models = {}
self.quantiles = quantiles
self.rmse_model = None
super().__init__()

def fit(self, X: pd.DataFrame, y: pd.Series) -> BaseEstimator:
"""
Expand Down
1 change: 1 addition & 0 deletions src/sageworks/algorithms/dataframe/residuals_calculator.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,7 @@ def __init__(
self.endpoint = endpoint # Use this endpoint for inference if provided
self.X = None
self.y = None
super().__init__()

def fit(self, X: pd.DataFrame, y: pd.Series) -> BaseEstimator:
"""
Expand Down
1 change: 1 addition & 0 deletions src/sageworks/algorithms/dataframe/target_gradients.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@ def __init__(self, n_neighbors: int = 2):
self.knn = KNeighborsRegressor(n_neighbors=self.n_neighbors, algorithm="ball_tree", metric="euclidean")
self.X = None
self.y = None
super().__init__()

def fit(self, X: pd.DataFrame, y: pd.Series) -> BaseEstimator:
"""
Expand Down

0 comments on commit 5d0cb2f

Please sign in to comment.