Updated image_test to work with newer versions of keras_preprocessing

PiperOrigin-RevId: 311266601
Change-Id: I0cbcea629b4fff04c50628432515a8766dc10ec8
This commit is contained in:
Tomer Kaftan 2020-05-12 22:19:52 -07:00 committed by TensorFlower Gardener
parent 75507c75e8
commit e71443f838

View File

@ -146,8 +146,7 @@ class TestImage(keras_parameterized.TestCase):
generator = preprocessing_image.ImageDataGenerator(
data_format='unknown')
generator = preprocessing_image.ImageDataGenerator(
zoom_range=(2, 2))
generator = preprocessing_image.ImageDataGenerator(zoom_range=(2., 2.))
def test_image_data_generator_fit(self):
generator = preprocessing_image.ImageDataGenerator(