Force conversion to Tensor of inference variables

han-ol · han-ol · commit c99fd011990e · 2025-03-14T21:21:26.000+01:00
diff --git a/bayesflow/approximators/continuous_approximator.py b/bayesflow/approximators/continuous_approximator.py
@@ -120,6 +120,8 @@ def compute_metrics(
             else:
                 inference_conditions = keras.ops.concatenate([inference_conditions, summary_outputs], axis=-1)
 
+        # Force a conversion to Tensor
+        inference_variables = keras.tree.map_structure(keras.ops.convert_to_tensor, inference_variables)
         inference_metrics = self.inference_network.compute_metrics(
             inference_variables, conditions=inference_conditions, stage=stage
         )

Original file line number	Diff line number	Diff line change
`@@ -120,6 +120,8 @@ def compute_metrics(`
`120`	`120`	`else:`
`121`	`121`	`inference_conditions = keras.ops.concatenate([inference_conditions, summary_outputs], axis=-1)`
`122`	`122`
	`123`	`+ # Force a conversion to Tensor`
	`124`	`+ inference_variables = keras.tree.map_structure(keras.ops.convert_to_tensor, inference_variables)`
`123`	`125`	`inference_metrics = self.inference_network.compute_metrics(`
`124`	`126`	`inference_variables, conditions=inference_conditions, stage=stage`
`125`	`127`	`)`