Update intervention_builder.py

pyciemss/integration_utils/intervention_builder.py

@@ -62,7 +62,7 @@ def intervention_generator(
 def combine_interventions(
     interventions: List[Dict[torch.Tensor, Dict[str, Intervention]]]
 ) -> Dict[torch.Tensor, Dict[str, Intervention]]:
-    static_parameter_interventions: Dict[float, Dict[str, Intervention]] = {}
+    static_parameter_interventions: Dict[torch.Tensor, Dict[str, Intervention]] = {}
     for intervention in interventions:
         for key, value in intervention.items():
             if key in static_parameter_interventions: