Fix bug introduced in Keras 3

philipperemy · Aug 13, 2024 · 1f4355c · 1f4355c
1 parent f612f6e
commit 1f4355c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tcn/tcn.py b/tcn/tcn.py
@@ -315,7 +315,10 @@ def build(self, input_shape):
         else:
             self.output_slice_index = -1  # causal case.
         self.slicer_layer = Lambda(lambda tt: tt[:, self.output_slice_index, :], name='Slice_Output')
-        self.slicer_layer.build(self.build_output_shape.as_list())
+        try:
+            self.slicer_layer.build(self.build_output_shape.as_list())
+        except AttributeError:
+            self.slicer_layer.build(self.build_output_shape)
 
     def compute_output_shape(self, input_shape):
         """