adding start_time_param_value

DARPA-ASKEM · Jul 17, 2024 · 477f968 · 477f968
1 parent 69c178f
commit 477f968
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/service/models/operations/optimize.py b/service/models/operations/optimize.py
@@ -10,6 +10,7 @@
 from pyciemss.integration_utils.intervention_builder import (
  param_value_objective,
  start_time_objective,
+ start_time_param_value_objective,
 )
 
 from pyciemss.ouu.qoi import obs_nday_average_qoi, obs_max_qoi
@@ -116,7 +117,7 @@ def gen_pyciemss_args(self, job_id):
  param_name=self.optimize_interventions.param_names,
  param_value=param_value,
  )
- else:
+ if intervention_type == "start_time":
  assert self.optimize_interventions.param_values is not None
  param_value = [
  torch.tensor(value)
@@ -126,6 +127,10 @@ def gen_pyciemss_args(self, job_id):
  param_name=self.optimize_interventions.param_names,
  param_value=param_value,
  )
+ if intervention_type == "start_time_param_value":
+ optimize_interventions = start_time_param_value_objective(
+ param_name=self.optimize_interventions.param_names
+ )
 
  extra_options = self.extra.dict()
  inferred_parameters = fetch_inferred_parameters(