fix (nn/convxd): typo fix on conditional

Xilinx · Sep 4, 2024 · 9de92f2 · 9de92f2
1 parent bc703e5
commit 9de92f2
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/brevitas/nn/quant_conv.py b/src/brevitas/nn/quant_conv.py
@@ -46,8 +46,8 @@ def __init__(
  dtype: Optional[torch.dtype] = None,
  **kwargs) -> None:
  # avoid an init error in the super class by setting padding to 0
- if padding_mode == 'zeros' and padding == 'same' and stride > 1 if isinstance(
- stride, int) else any(map(lambda x: x > 1, stride)):
+ if padding_mode == 'zeros' and padding == 'same' and (stride > 1 if isinstance(
+ stride, int) else any(map(lambda x: x > 1, stride))):
  padding = 0
  is_same_padded_strided = True
  else:
@@ -133,8 +133,8 @@ def __init__(
  dtype: Optional[torch.dtype] = None,
  **kwargs) -> None:
  # avoid an init error in the super class by setting padding to 0
- if padding_mode == 'zeros' and padding == 'same' and stride > 1 if isinstance(
- stride, int) else any(map(lambda x: x > 1, stride)):
+ if padding_mode == 'zeros' and padding == 'same' and (stride > 1 if isinstance(
+ stride, int) else any(map(lambda x: x > 1, stride))):
  padding = 0
  is_same_padded_strided = True
  else:
@@ -222,8 +222,8 @@ def __init__(
  dtype: Optional[torch.dtype] = None,
  **kwargs) -> None:
  # avoid an init error in the super class by setting padding to 0
- if padding_mode == 'zeros' and padding == 'same' and stride > 1 if isinstance(
- stride, int) else any(map(lambda x: x > 1, stride)):
+ if padding_mode == 'zeros' and padding == 'same' and (stride > 1 if isinstance(
+ stride, int) else any(map(lambda x: x > 1, stride))):
  padding = 0
  is_same_padded_strided = True
  else: