scvi.dataloaders.AnnTorchDataset.register_function

classmethod AnnTorchDataset.register_function(function_name, function)