def METHOD_NAME():
    adata = scvi.data.synthetic_iid()
    manager = generic_setup_adata_manager(adata, batch_key="batch")
    dataset = manager.create_torch_dataset()
    data = dataset[:10]
    assert isinstance(data, dict)
    assert REGISTRY_KEYS.X_KEY in data
    assert REGISTRY_KEYS.BATCH_KEY in data
    assert isinstance(data[REGISTRY_KEYS.X_KEY], np.ndarray)
    assert isinstance(data[REGISTRY_KEYS.BATCH_KEY], np.ndarray)
    assert data[REGISTRY_KEYS.X_KEY].dtype == np.float32
    assert data[REGISTRY_KEYS.BATCH_KEY].dtype == np.int64