Internal change

PiperOrigin-RevId: 480462272

Internal change
PiperOrigin-RevId: 480462272
3347c155 · Rebecca Chen · A. Unique TensorFlower · baf94acc · 3347c155
Commit 3347c155 authored Oct 11, 2022 by Rebecca Chen Committed by A. Unique TensorFlower Oct 11, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/projects/assemblenet/configs/assemblenet.py official/projects/assemblenet/configs/assemblenet.py +1 -1

No files found.
--- a/official/projects/assemblenet/configs/assemblenet.py
+++ b/official/projects/assemblenet/configs/assemblenet.py
@@ -61,7 +61,7 @@ def flat_lists_to_blocks(model_structures, model_edge_weights):
    if node[0] < 0:
      block = BlockSpec(level=node[0], temporal_dilation=node[1])
    else:
-      block = BlockSpec(
+      block = BlockSpec(  # pytype: disable=wrong-arg-types
          level=node[0],
          input_blocks=node[1],
          num_filters=node[2],