Remove .shape reference for autograph [skip ci]

Author: stefanradev93

bayesflow/amortizers.py

@@ -1174,7 +1174,7 @@ def _compute_condition(self, input_dict, **kwargs):
         local_summaries, global_summaries = self._get_local_global(input_dict, **kwargs)
 
         # At this point, add globals as conditions
-        num_locals = local_summaries.shape[1]
+        num_locals = tf.shape(local_summaries)[1]
 
         # Add hyper parameters as conditions:
         # p(local_n | data_n, hyper)