typing fix

Author: Atticus1806

nn/rec.py

@@ -11,7 +11,7 @@ class LSTM(_Rec):
   """
   LSTM operating on a sequence. returns (output, final_state) tuple, where final_state is (h,c).
   """
-  def __init__(self, n_out: int, zoneout: Optional[bool] = False, **kwargs):
+  def __init__(self, n_out: int, zoneout: bool = False, **kwargs):
     if zoneout:
       super().__init__(n_out=n_out, unit="zoneoutlstm", **kwargs)
     else:
@@ -30,7 +30,7 @@ class LSTMStep(_Rec):
   """
   default_name = "lstm"  # make consistent to LSTM
 
-  def __init__(self, n_out: int, zoneout: Optional[bool] = False, **kwargs):
+  def __init__(self, n_out: int, zoneout: bool = False, **kwargs):
     if zoneout:
       super().__init__(n_out=n_out, unit="zoneoutlstm", **kwargs)
     else: