cc50b0085728cda00df44909a971384e7c585f3f,tests/framework/test_model.py,ModelTest,_test_fitting,#ModelTest#Any#Any#,118

Before Change

        for log in logs:
            keys = ["epoch", "lr", "loss", "some_metric_1", "some_metric_2", "val_loss", "val_some_metric_1", "val_some_metric_2"]
            self.assertCountEqual(log.keys(), keys)
            self.assertEqual(log["some_metric_1"], 1.)
            self.assertEqual(log["some_metric_2"], 2.)
            self.assertEqual(log["val_some_metric_1"], 1.)
            self.assertEqual(log["val_some_metric_2"], 2.)

After Change

        for log in logs:
            keys = ["epoch", "lr", "loss"] + self.metrics_names + ["val_loss"] + ["val_" + metric_name for metric_name in self.metrics_names]
            self.assertCountEqual(log.keys(), keys)
            for metric_name, metric_value in zip(self.metrics_names, self.metrics_values):
                with self.subTest(metric_name=metric_name):
                    self.assertEqual(log[metric_name], metric_value)
                    self.assertEqual(log["val_" + metric_name], metric_value)

        method_calls = self.mock_callback.method_calls
        self.assertEqual(method_calls.index(call.on_train_begin([])), 2)
        self.assertIn(call.set_model(self.model), method_calls[:2])
        self.assertIn(call.set_params(params), method_calls[:2])

Instances