Update dependency_classifier.py

dependency_classifier.py

@@ -1,4 +1,4 @@
-from typing import override
+#from typing import override
 from copy import deepcopy
 
 import numpy as np
@@ -127,7 +127,7 @@ class DependencyHead(DependencyHeadBase):
     Basic UD syntax specialization that predicts single edge for each token.
     """
 
-    @override
+    #@override
     def predict_arcs(
         self,
         s_arc: Tensor,           # [batch_size, seq_len, seq_len]
@@ -191,7 +191,7 @@ class DependencyHead(DependencyHeadBase):
         return pred_arcs
 
     @staticmethod
-    @override
+    #@override
     def calc_arc_loss(
         s_arc: Tensor,         # [batch_size, seq_len, seq_len]
         gold_arcs: LongTensor  # [n_arcs, 4]
@@ -205,7 +205,7 @@ class MultiDependencyHead(DependencyHeadBase):
     Enhanced UD syntax specialization that predicts multiple edges for each token.
     """
 
-    @override
+    #@override
     def predict_arcs(
         self,
         s_arc: Tensor,           # [batch_size, seq_len, seq_len]
@@ -218,7 +218,7 @@ class MultiDependencyHead(DependencyHeadBase):
         return arc_probs.round().long()
 
     @staticmethod
-    @override
+    #@override
     def calc_arc_loss(
         s_arc: Tensor,         # [batch_size, seq_len, seq_len]
         gold_arcs: LongTensor  # [n_arcs, 4]