---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
Cell In[28], line 1
----> 1 train_dataset.save_to_disk("test")

File /usr/local/lib/python3.10/dist-packages/datasets/arrow_dataset.py:1496, in Dataset.save_to_disk(self, dataset_path, fs, max_shard_size, num_shards, num_proc, storage_options)
   1494         json.dumps(state["_format_kwargs"][k])
   1495     except TypeError as e:
-> 1496         raise TypeError(
   1497             str(e) + f"\nThe format kwargs must be JSON serializable, but key '{k}' isn't."
   1498         ) from None
   1499 # Get json serializable dataset info
   1500 dataset_info = asdict(self._info)

TypeError: Object of type function is not JSON serializable
The format kwargs must be JSON serializable, but key 'transform' isn't.