Skip to content
5 changes: 5 additions & 0 deletions openml/runs/__init__.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
# License: BSD 3-Clause

import importlib.util

from .functions import (
delete_run,
get_run,
Expand All @@ -15,6 +17,9 @@
from .run import OpenMLRun
from .trace import OpenMLRunTrace, OpenMLTraceIteration

if importlib.util.find_spec("openml_sklearn"):
import openml_sklearn # noqa: F401

__all__ = [
"OpenMLRun",
"OpenMLRunTrace",
Expand Down