Skip to content

Commit 480bbb2

Browse files
committed
Use keras_utils when testing instead of TensorFlow private API
1 parent a64e05b commit 480bbb2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow_addons/layers/tests/deformable_conv2d_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
import pytest
1818
import numpy as np
1919
import tensorflow as tf
20-
from tensorflow.python.keras.utils import conv_utils
20+
import tensorflow_addons.utils.keras_utils as conv_utils
2121
from tensorflow_addons.layers.deformable_conv2d import DeformableConv2D
2222

2323

0 commit comments

Comments
 (0)