Update i6_models/parts/frontend/generic_frontend.py

Co-authored-by: SimBe195 <[email protected]>
rwth-i6 · Nov 10, 2023 · aea6122 · aea6122
1 parent dde2533
commit aea6122
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/i6_models/parts/frontend/generic_frontend.py b/i6_models/parts/frontend/generic_frontend.py
@@ -46,13 +46,13 @@ class GenericFrontendV1Config(ModelConfiguration):
 
     in_features: int
     layer_ordering: Sequence[FrontendLayerType]
-    conv_kernel_sizes: Optional[Sequence[Union[int, Tuple[int, int]]]]
-    conv_strides: Optional[Sequence[Union[int, Tuple[int, int]]]]
-    conv_paddings: Optional[Sequence[Union[int, Tuple[int, int]]]]
-    conv_out_dims: Optional[Sequence[Union[int, Tuple[int, int]]]]
-    pool_kernel_sizes: Optional[Sequence[Union[int, Tuple[int, int]]]]
-    pool_strides: Optional[Sequence[Union[int, Tuple[int, int]]]]
-    pool_paddings: Optional[Sequence[Union[int, Tuple[int, int]]]]
+    conv_kernel_sizes: Optional[Sequence[Tuple[int, int]]]
+    conv_strides: Optional[Sequence[Tuple[int, int]]]
+    conv_paddings: Optional[Sequence[Tuple[int, int]]]
+    conv_out_dims: Optional[Sequence[int]]
+    pool_kernel_sizes: Optional[Sequence[Tuple[int, int]]]
+    pool_strides: Optional[Sequence[Tuple[int, int]]]
+    pool_paddings: Optional[Sequence[Tuple[int, int]]]
     activations: Optional[Sequence[Union[nn.Module, Callable[[torch.Tensor], torch.Tensor]]]]
     out_features: int