approved suggestion

dan/predict/attention.py

@@ -39,8 +39,7 @@ def compute_coverage(text: str, max_value: float, offset: int, attentions):
     :param offset: Offset value to get the relevant part of text piece
     :param attentions: Attention weights of size (n_char, feature_height, feature_width)
     """
-    height = attentions.shape[1]
-    width = attentions.shape[2]
+    _, height, width = attentions.shape
 
     # blank vector to accumulate weights for the current text
     coverage_vector = np.zeros((height, width))