code: def test_fit(ds_type, num_batches):
      dm_ref = TensorDataModule(torch.randn(500, 5), ds_type=ds_type)
      d = torchdrift.detectors.KernelMMDDriftDetector()
      torchdrift.utils.fit(
          dm_ref.default_dataloader(batch_size=10),
          torch.nn.Identity(),
          [torch.nn.Identity(), d],
          num_batches=num_batches,
          device="cpu",
      )
  
file_path: TorchDrift/test/test_utils.py
imports: import pytest
  import torchdrift
  import torch
  import torch.utils.data
label_desc: Use image node_id 1 for calling a global function with example usage: test_fit(ds_type, num_batches) without return types