update test

DARPA-ASKEM · Feb 3, 2025 · 5a35099 · 5a35099
1 parent 0b46114
commit 5a35099
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/service/models/operations/optimize.py b/service/models/operations/optimize.py
@@ -160,7 +160,6 @@ class InterventionObjective(BaseModel):
         example="param_value",
     )
     param_name: str
-    # TODO Do we need param_value and start_time or can we use init?
     param_value: Optional[Optional[float]] = None
     start_time: Optional[float] = None
     time_objective_function: Optional[InterventionObjectiveFunction] = None

diff --git a/tests/examples/optimize/input/request.json b/tests/examples/optimize/input/request.json
@@ -4,12 +4,18 @@
 	"model_config_id": "sidarthe",
 	"optimize_interventions": [{
 		"intervention_type": "param_value",
-		"objective_function_option": "lower_bound",
-		"start_time": [2],
-		"param_names": ["beta"],
-		"param_values": [0.02],
-		"initial_guess": [0],
-		"relative_importance": 1
+		"param_name": "beta",
+		"start_time": 2,
+		"param_value": 0.02,
+		"time_objective_function": "initial_guess",
+		"parameter_objective_function": "initial_guess",
+		"relative_importance": 5,
+		"param_value_initial_guess": 0.02,
+		"parameter_value_lower_bound": 0,
+		"parameter_value_upper_bound": 0.08,
+		"start_time_initial_guess": 1,
+		"start_time_lower_bound": 0,
+		"start_time_upper_bound": 50
 	}],
 	"timespan": {
 		"start": 0,