Skip to content

Commit 43125ad

Browse files
committed
Rebasing
2 parents 840f4fd + 02653c2 commit 43125ad

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

openml/extensions/sklearn/extension.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,6 @@ def _min_dependency_str(cls, sklearn_version: str) -> str:
223223
Returns
224224
-------
225225
str
226-
227226
"""
228227
if LooseVersion(sklearn_version) >= "0.24":
229228
# assumption: 0.24 onwards sklearn should contain a _min_dependencies.py file with
@@ -812,7 +811,13 @@ def _serialize_model(self, model: Any) -> OpenMLFlow:
812811
return flow
813812

814813
def _get_dependencies(self) -> str:
815-
dependencies = self._min_dependency_str(sklearn.__version__)
814+
dependencies = "\n".join(
815+
[
816+
self._format_external_version("sklearn", sklearn.__version__,),
817+
"numpy>=1.6.1",
818+
"scipy>=0.9",
819+
]
820+
)
816821
return dependencies
817822

818823
def _get_tags(self) -> List[str]:

0 commit comments

Comments
 (0)