[FIX] bad argument name in tf_reshape

wbenbihi · Aug 20, 2022 · 60da17a · 60da17a
1 parent 0cef451
commit 60da17a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hourglass_tensorflow/utils/tf.py b/hourglass_tensorflow/utils/tf.py
@@ -165,7 +165,7 @@ def tf_expand_bbox(
         bottom_right_y + height * (bbox_factor - 1.0),
     )
     return tf.cast(
-        tf_reshape_slice([new_tl_x, new_tl_y, new_br_x, new_br_y], dim=2, **kwargs),
+        tf_reshape_slice([new_tl_x, new_tl_y, new_br_x, new_br_y], shape=2, **kwargs),
         dtype=tf.int32,
     )