trchdiffeq

Author: bartzbeielstein

pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "spotpython"
-version = "0.22.2"
+version = "0.22.3"
 authors = [
   { name="T. Bartz-Beielstein", email="tbb@bartzundbartz.de" }
 ]
@@ -51,6 +51,7 @@ dependencies = [
   "tabulate",
   "tensorboard",
   "torch",
+  "torchdiffeq",
   "torch-tb-profiler",
   "torchmetrics",
   "torchvision",

src/spotpython/utils/file.py

@@ -95,7 +95,7 @@ def get_experiment_filename(PREFIX) -> str:
     return filename
 
 
-def load_result(PREFIX) -> None:
+def load_result(PREFIX) -> tuple:
     """Loads the result from a pickle file with the name
     PREFIX + "_res.pkl".
     This is the standard filename for the result file,
@@ -105,6 +105,9 @@ def load_result(PREFIX) -> None:
     Args:
         PREFIX (str): Prefix of the experiment.
 
+    Returns:
+        (spot_tuner, fun_control, design_control, surrogate_control, optimizer_control): Tuple of objects.
+
     Examples:
         >>> from spotpython.utils.file import load_result
         >>> load_result("branin")