Updated image_test to work with newer versions of keras_preprocessing
PiperOrigin-RevId: 311266601 Change-Id: I0cbcea629b4fff04c50628432515a8766dc10ec8
This commit is contained in:
parent
75507c75e8
commit
e71443f838
@ -146,8 +146,7 @@ class TestImage(keras_parameterized.TestCase):
|
|||||||
generator = preprocessing_image.ImageDataGenerator(
|
generator = preprocessing_image.ImageDataGenerator(
|
||||||
data_format='unknown')
|
data_format='unknown')
|
||||||
|
|
||||||
generator = preprocessing_image.ImageDataGenerator(
|
generator = preprocessing_image.ImageDataGenerator(zoom_range=(2., 2.))
|
||||||
zoom_range=(2, 2))
|
|
||||||
|
|
||||||
def test_image_data_generator_fit(self):
|
def test_image_data_generator_fit(self):
|
||||||
generator = preprocessing_image.ImageDataGenerator(
|
generator = preprocessing_image.ImageDataGenerator(
|
||||||
|
Loading…
Reference in New Issue
Block a user