def adata_parameterized(request):
    return gen_adata(shape=(200, 300), X_type=request.param)
def tearDown(self):
        if hasattr(self, "rng_state"):
            torch.set_rng_state(self.rng_state)