Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
All flow dependencies for sklearn>0.24 will change now
  • Loading branch information
Neeratyoy committed Feb 16, 2021
commit 844257695a59c37f0eb9d6d618304723b991bd4d
9 changes: 1 addition & 8 deletions openml/extensions/sklearn/extension.py
Original file line number Diff line number Diff line change
Expand Up @@ -804,20 +804,13 @@ def _serialize_model(self, model: Any) -> OpenMLFlow:
tags=tags,
extension=self,
language="English",
# TODO fill in dependencies!
dependencies=dependencies,
)

return flow

def _get_dependencies(self) -> str:
dependencies = "\n".join(
[
self._format_external_version("sklearn", sklearn.__version__,),
"numpy>=1.6.1",
"scipy>=0.9",
]
)
dependencies = self._min_dependency_str(sklearn.__version__)
return dependencies

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