diff --git a/docs/source/interfaces.ipynb b/docs/source/interfaces.ipynb index cdd1700d..b9e70833 100644 --- a/docs/source/interfaces.ipynb +++ b/docs/source/interfaces.ipynb @@ -2539,7 +2539,7 @@ "name": "stderr", "output_type": "stream", "text": [ - " 88%|██████████████████████████████████████████▍ | 53/60 [00:07<00:00, 9.47it/s]" + " 75%|████████████████████████████████████ | 45/60 [00:06<00:01, 9.23it/s]" ] }, { @@ -2549,11 +2549,11 @@ "Optimal policy: tensor([0.2032], dtype=torch.float64)\n", "{'policy': tensor([0.2032], dtype=torch.float64), 'OptResults': message: ['requested number of basinhopping iterations completed successfully']\n", " success: True\n", - " fun: 0.2111072303557351\n", - " x: [ 1.389e-01]\n", + " fun: 0.14687910929135808\n", + " x: [ 2.032e-01]\n", " nit: 5\n", " minimization_failures: 4\n", - " nfev: 53\n", + " nfev: 46\n", " lowest_optimization_result: message: Optimization terminated successfully.\n", " success: True\n", " status: 1\n", diff --git a/pyciemss/ouu/ouu.py b/pyciemss/ouu/ouu.py index 6f0315c5..09f1cd50 100644 --- a/pyciemss/ouu/ouu.py +++ b/pyciemss/ouu/ouu.py @@ -8,7 +8,6 @@ import torch from chirho.dynamical.handlers.solver import TorchDiffEq from chirho.interventional.ops import Intervention - from scipy.optimize import basinhopping from pyciemss.integration_utils.intervention_builder import (