Include solver options for optimize

DARPA-ASKEM · Mar 21, 2024 · 06341eb · 06341eb
1 parent 49dc469
commit 06341eb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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):