From 06341ebeed32edfe8ec4e043f43566065779fb3f Mon Sep 17 00:00:00 2001 From: Five Grant <5@fivegrant.com> Date: Thu, 21 Mar 2024 10:41:33 -0500 Subject: [PATCH] Include solver options for optimize --- service/models/operations/optimize.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/service/models/operations/optimize.py b/service/models/operations/optimize.py index e8e2b4d..d40c08d 100644 --- a/service/models/operations/optimize.py +++ b/service/models/operations/optimize.py @@ -1,6 +1,6 @@ from __future__ import annotations -from typing import ClassVar, Dict, List, Optional +from typing import Any, ClassVar, Dict, List, Optional import numpy as np import torch @@ -71,6 +71,7 @@ class OptimizeExtra(BaseModel): is_minimized: bool = True alpha: float = 0.95 solver_method: str = "dopri5" + solver_options: Dict[str, Any] = {} class Optimize(OperationRequest):