diff --git a/wekws/model/mdtc.py b/wekws/model/mdtc.py index 87f1558..090dde5 100644 --- a/wekws/model/mdtc.py +++ b/wekws/model/mdtc.py @@ -217,7 +217,7 @@ class MDTC(nn.Module): super(MDTC, self).__init__() assert kernel_size % 2 == 1 self.kernel_size = kernel_size - assert causal == True, "we now only support causal mdtc" + assert causal is True, "we now only support causal mdtc" self.causal = causal self.preprocessor = TCNBlock(in_channels, res_channels,