Merge pull request #1 from dimx9173/dev/philipperemy#256

Fix issue#256
dimx9173 · Aug 5, 2024 · 042b9df · 042b9df
2 parents fc6ffd3 + 0b8928d
commit 042b9df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tcn/tcn.py b/tcn/tcn.py
@@ -315,7 +315,7 @@ 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())
+        self.slicer_layer.build(list(self.build_output_shape))
 
     def compute_output_shape(self, input_shape):
         """