Error en el test_generator

[dasadeco]

En el código del tema 11 parece que hay un error en este repo, que sin embargo es correcto en el libro:

test_generator =  test_datagen.flow_from_directory(validation_dir,
                                                         batch_size=20,
                                                         class_mode  = 'binary',
                                                         target_size = (150, 150))

Supongo que debería ser 'test_dir'