def test_set_transform_on_empty_config(self) -> None: """Test set_transform.""" config = Config() config.set_transform([ { "name": "Some transform1", "params": { "param12": True } }, { "name": "SquadV1", "params": { "param1": True } }, { "name": "Some transform2", "params": { "param123": True } }, ], ) self.assertIsNone(config.evaluation) self.assertIsNone(config.quantization)
def test_set_transform(self) -> None: """Test set_transform.""" config = Config(self.predefined_config) config.set_transform([ { "name": "Some transform1", "params": { "param12": True } }, { "name": "SquadV1", "params": { "param1": True } }, { "name": "Some transform2", "params": { "param123": True } }, ], ) self.assertEqual( { "SquadV1": { "param1": True }, }, config.evaluation.accuracy.postprocess.transform, ) self.assertEqual( ["Some transform1", "Some transform2"], list(config.quantization.calibration.dataloader.transform.keys()), ) self.assertEqual( ["Some transform1", "Some transform2"], list(config.evaluation.accuracy.dataloader.transform.keys()), ) self.assertEqual( ["Some transform1", "Some transform2"], list(config.evaluation.performance.dataloader.transform.keys()), )