def test_dataloader_serializer(self) -> None: """Test dataloader config serializer.""" data = { "last_batch": "rollover", "batch_size": 2, "dataset": { "TestDataset": { "dataset_param": "/some/path", "bool_param": True, "list_param": ["item1", "item2"], }, }, "transform": { "TestTransform": { "shape": [1000, 224, 224, 3], "some_op": True }, "AnotherTestTransform": { "shape": [10, 299, 299, 3], "some_op": False }, }, "filter": { "LabelBalance": { "size": 1 }, }, } dataloader = Dataloader(data) self.assertDictEqual( dataloader.serialize(), { "last_batch": "rollover", "batch_size": 2, "dataset": { "TestDataset": { "dataset_param": "/some/path", "bool_param": True, "list_param": ["item1", "item2"], }, }, "transform": { "TestTransform": { "shape": [1000, 224, 224, 3], "some_op": True }, "AnotherTestTransform": { "shape": [10, 299, 299, 3], "some_op": False, }, }, "filter": { "LabelBalance": { "size": 1 }, }, }, )
def test_dataloader_serializer_defaults(self) -> None: """Test dataloader config serializer defaults.""" dataloader = Dataloader() self.assertDictEqual( dataloader.serialize(), { "batch_size": 1, }, )