fight with CI

elitcloud · Jun 13, 2019 · 5137fc2 · 5137fc2
1 parent 87fe1d8
commit 5137fc2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/elit/component/sem/biaffine_parser.py b/elit/component/sem/biaffine_parser.py
@@ -216,8 +216,8 @@ def biaffine(self, dep_arc, dep_rel, head_arc, head_rel, mask, arc_targets, rel_
                                           (self._vocab.rel_size, seq_len * seq_len * batch_size))
         # rel_size x (#head x #dep x batch_size)
         if is_train or arc_targets is not None:
-            mask_rel: nd.NDArray = reshape_fortran(nd.array(mask * arc_targets),
-                                                   (1, seq_len * seq_len * batch_size))
+            mask_rel = reshape_fortran(nd.array(mask * arc_targets),
+                                       (1, seq_len * seq_len * batch_size))  # type: nd.NDArray
             flat_rel_preds = flat_rel_logits.argmax(0)
             flat_rel_target = nd.array(reshape_fortran(rel_targets, (1, seq_len * seq_len * batch_size))).squeeze(
                 axis=0)