afd5dc51f603cdf30e1991993ba4f06f1f385fc9,tests/framework/model/test_model.py,ModelTest,test_fitting_with_no_coloring,#ModelTest#,296

Before Change

        valid_generator = some_data_tensor_generator(ModelTest.batch_size)

        // Capture the output
        self.test_out = io.StringIO()
        self.original_output = sys.stdout
        sys.stdout = self.test_out

        _ = self.model.fit_generator(train_generator,
                                     valid_generator,
                                     epochs=ModelTest.epochs,

After Change

                                         callbacks=[self.mock_callback],
                                         progress_options=dict(coloring={"invalid_name": "A COLOR"}))

    def test_fitting_with_no_coloring(self):
        train_generator = some_data_tensor_generator(ModelTest.batch_size)
        valid_generator = some_data_tensor_generator(ModelTest.batch_size)

        self._capture_output()

        _ = self.model.fit_generator(train_generator,
                                     valid_generator,
                                     epochs=ModelTest.epochs,

Instances