add import of ml-test-functions

Author: SimonBlanke

src/surfaces/test_functions/machine_learning/__init__.py

@@ -36,7 +36,7 @@ def _check_sklearn():
         SVMRegressorFunction,
     )
 
-    # Time-series functions
+    # Time-series functions (sklearn-based)
     from .timeseries import (
         # Forecasting
         GradientBoostingForecasterFunction,
@@ -100,6 +100,26 @@ def _check_sklearn():
         RandomForestImageClassifierFunction,
     ]
 
+    # sktime-based time-series functions (require sktime)
+    try:
+        from .timeseries import ExpSmoothingForecasterFunction, TSForestClassifierFunction
+
+        __all__.extend(
+            [
+                "ExpSmoothingForecasterFunction",
+                "TSForestClassifierFunction",
+            ]
+        )
+        machine_learning_functions.extend(
+            [
+                ExpSmoothingForecasterFunction,
+                TSForestClassifierFunction,
+            ]
+        )
+        _HAS_SKTIME = True
+    except ImportError:
+        _HAS_SKTIME = False
+
     # CNN image classifiers (require tensorflow)
     try:
         from .image import SimpleCNNClassifierFunction, DeepCNNClassifierFunction
@@ -120,6 +140,16 @@ def _check_sklearn():
     except ImportError:
         _HAS_TENSORFLOW = False
 
+    # XGBoost image classifier (requires xgboost)
+    try:
+        from .image import XGBoostImageClassifierFunction
+
+        __all__.append("XGBoostImageClassifierFunction")
+        machine_learning_functions.append(XGBoostImageClassifierFunction)
+        _HAS_XGBOOST = True
+    except ImportError:
+        _HAS_XGBOOST = False
+
 else:
     __all__ = []
     machine_learning_functions = []