Skip to content

Commit

Permalink
Relax upper version bounds on snapml
Browse files Browse the repository at this point in the history
Signed-off-by: Avi Shinnar <shinnar@us.ibm.com>
  • Loading branch information
shinnar committed Feb 2, 2024
1 parent 6af7644 commit d29b200
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 1 deletion.
21 changes: 21 additions & 0 deletions lale/lib/snapml/snap_boosting_machine_regressor.py
Original file line number Diff line number Diff line change
Expand Up @@ -362,4 +362,25 @@ def predict(self, X, **predict_params):
}
)

if snapml_version is not None and snapml_version >= version.Version("1.14"):
SnapBoostingMachineRegressor = SnapBoostingMachineRegressor.customize_schema(
alpha={
"type": "number",
"minimum": 0.0,
"exclusiveMinimum": True,
"minimumForOptimizer": 1e-10,
"maximumForOptimizer": 1.0,
"default": 0.5,
"description": "Quantile used when 'objective = quantile'.",
},
min_h_quantile={
"type": "number",
"minimumForOptimizer": 0.0,
"maximumForOptimizer": 1.0,
"default": 0.0,
"description": "Regularization term for quantile regression.",
},
set_as_available=True,
)

lale.docstrings.set_docstrings(SnapBoostingMachineRegressor)
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@
"mystic",
"xgboost<2.1.0",
"lightgbm<4.4.0",
"snapml>=1.7.0rc3,<1.12.0",
"snapml>=1.7.0rc3,<1.16.0",
"liac-arff>=2.4.0",
"tensorflow>=2.4.0,<=2.13.0",
"smac<=0.10.0",
Expand Down

0 comments on commit d29b200

Please sign in to comment.