removing unintended []

DARPA-ASKEM · Feb 27, 2024 · 38251a9 · 38251a9
1 parent 5016e23
commit 38251a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/service/models/converters.py b/service/models/converters.py
@@ -14,7 +14,7 @@ def convert_to_static_interventions(interventions):
 def convert_optimize_to_static_interventions(interventions):
  static_interventions = defaultdict(dict)
  for i in interventions:
- static_interventions[i.timestep] = torch.tensor([i.name])
+ static_interventions[i.timestep] = torch.tensor(i.name)
  return static_interventions