Hi, thanks for creating this amazing library!

There seems to be a small issue with the lance.torch.data.LanceDataset in the to_tensor_fn kwarg of the initializer. Its type hint is:

to_tensor_fn: Optional[
    callable[[pa.RecordBatch], Union[dict[str, torch.Tensor], torch.Tensor]]
] = None,

However, it is called here with an additional kwarg:

batch = self._to_tensor_fn(batch, hf_converter=self._hf_converter)