fix float32

arrjon · arrjon · commit caa2d67ec4f9 · 2025-09-10T18:54:15.000+02:00
diff --git a/bayesflow/approximators/continuous_approximator.py b/bayesflow/approximators/continuous_approximator.py
@@ -706,7 +706,8 @@ def compute_prior_score_pre(_samples: Tensor) -> Tensor:
             )
             prior_score = compute_prior_score(adapted_samples)
             for key in adapted_samples:
-                prior_score[key] = prior_score[key]
+                if isinstance(prior_score[key], np.ndarray):
+                    prior_score[key] = prior_score[key].astype("float32")
                 if len(log_det_jac) > 0:
                     prior_score[key] += log_det_jac[key]
 

Original file line number	Diff line number	Diff line change
`@@ -706,7 +706,8 @@ def compute_prior_score_pre(_samples: Tensor) -> Tensor:`
`706`	`706`	`)`
`707`	`707`	`prior_score = compute_prior_score(adapted_samples)`
`708`	`708`	`for key in adapted_samples:`
`709`		`- prior_score[key] = prior_score[key]`
	`709`	`+ if isinstance(prior_score[key], np.ndarray):`
	`710`	`+ prior_score[key] = prior_score[key].astype("float32")`
`710`	`711`	`if len(log_det_jac) > 0:`
`711`	`712`	`prior_score[key] += log_det_jac[key]`
`712`	`713`