From 0765c5968d199e176c47aa9a7024762c5eb18c51 Mon Sep 17 00:00:00 2001 From: Tom Szendrey Date: Tue, 27 Feb 2024 14:47:17 -0500 Subject: [PATCH] I believe I had name and timestep reversed --- service/models/converters.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/service/models/converters.py b/service/models/converters.py index be644e9..ba79dad 100644 --- 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.name] = torch.tensor(i.timestep) return static_interventions