We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 840f4fd + 02653c2 commit 43125adCopy full SHA for 43125ad
1 file changed
openml/extensions/sklearn/extension.py
@@ -223,7 +223,6 @@ def _min_dependency_str(cls, sklearn_version: str) -> str:
223
Returns
224
-------
225
str
226
-
227
"""
228
if LooseVersion(sklearn_version) >= "0.24":
229
# assumption: 0.24 onwards sklearn should contain a _min_dependencies.py file with
@@ -812,7 +811,13 @@ def _serialize_model(self, model: Any) -> OpenMLFlow:
812
811
return flow
813
814
def _get_dependencies(self) -> str:
815
- dependencies = self._min_dependency_str(sklearn.__version__)
+ dependencies = "\n".join(
+ [
816
+ self._format_external_version("sklearn", sklearn.__version__,),
817
+ "numpy>=1.6.1",
818
+ "scipy>=0.9",
819
+ ]
820
+ )
821
return dependencies
822
823
def _get_tags(self) -> List[str]:
0 commit comments