Fix data type issue in _compute_doppler_shifts, fixes #426 (#436)

Signed-off-by: Florian Euchner <[email protected]> Merged-by: Guillermo Marcus <[email protected]>
NVlabs · May 17, 2024 · 2cb12fd · 2cb12fd
1 parent 8ad32bc
commit 2cb12fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sionna/rt/solver_paths.py b/sionna/rt/solver_paths.py
@@ -4124,7 +4124,7 @@ def _compute_doppler_shifts(self, paths, paths_tmp, velocity):
         # Compute Doppler shift per path
         #[num_targets, num_sources, max_num_paths]
         doppler = tf.reduce_sum(velocity*k_diff, axis=-1)
-        doppler = tf.where(objects_mask, 0, doppler)
+        doppler = tf.where(objects_mask, tf.constant(0, doppler.dtype), doppler)
         doppler = tf.reduce_sum(doppler, axis=0)
         doppler /= self._scene.wavelength
         return doppler