Merge pull request #74 from vpratz/Development

Bugfix: check_tensor_sanity use of .numpy() only possible in eager mode.

bayesflow/helper_functions.py

@@ -29,13 +29,20 @@
 
 def check_tensor_sanity(tensor, logger):
  """Tests for the present of NaNs and Infs in a tensor."""
-
- if tf.reduce_any(tf.math.is_nan(tensor)):
- num_na = tf.reduce_sum(tf.cast(tf.math.is_nan(tensor), tf.int8)).numpy()
- logger.warn(f"Warning! Returned estimates contain {num_na} nan values!")
- if tf.reduce_any(tf.math.is_inf(tensor)):
- num_inf = tf.reduce_sum(tf.cast(tf.math.is_inf(tensor), tf.int8)).numpy()
- logger.warn(f"Warning! Returned estimates contain {num_inf} inf values!")
+ if tf.executing_eagerly():
+ if tf.reduce_any(tf.math.is_nan(tensor)):
+ num_na = tf.reduce_sum(tf.cast(tf.math.is_nan(tensor), tf.int8)).numpy()
+ logger.warn(f"Warning! Returned estimates contain {num_na} nan values!")
+ if tf.reduce_any(tf.math.is_inf(tensor)):
+ num_inf = tf.reduce_sum(tf.cast(tf.math.is_inf(tensor), tf.int8)).numpy()
+ logger.warn(f"Warning! Returned estimates contain {num_inf} inf values!")
+ else:
+ if tf.reduce_any(tf.math.is_nan(tensor)):
+ num_na = tf.reduce_sum(tf.cast(tf.math.is_nan(tensor), tf.int8))
+ tf.print("Warning! Returned estimates contain", num_na, "nan values!")
+ if tf.reduce_any(tf.math.is_inf(tensor)):
+ num_inf = tf.reduce_sum(tf.cast(tf.math.is_inf(tensor), tf.int8))
+ tf.print(f"Warning! Returned estimates contain", num_inf, "inf values!")
 
 
 def merge_left_into_right(left_dict, right_dict):