Skip to content

Commit 95ca126

Browse files
committed
fix: remove unnecessary covert_to_tensor call
1 parent 3455ce1 commit 95ca126

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bayesflow/experimental/diffusion_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -137,9 +137,9 @@ def validate(self):
137137
if self._log_snr_min >= self._log_snr_max:
138138
raise ValueError("min_log_snr must be less than max_log_snr.")
139139
for training in [True, False]:
140-
if not ops.isfinite(self.get_log_snr(ops.convert_to_tensor(0), training=training)):
140+
if not ops.isfinite(self.get_log_snr(0.0, training=training)):
141141
raise ValueError("log_snr(0) must be finite.")
142-
if not ops.isfinite(self.get_log_snr(ops.convert_to_tensor(1), training=training)):
142+
if not ops.isfinite(self.get_log_snr(1.0, training=training)):
143143
raise ValueError("log_snr(1) must be finite.")
144144
if not ops.isfinite(self.get_t_from_log_snr(self._log_snr_max, training=training)):
145145
raise ValueError("t(0) must be finite.")

0 commit comments

Comments
 (0)