5b83fcf8d524cab79fbb3195653ae824fafc737c,test/priors/test_lkj_prior.py,TestLKJPrior,test_lkj_prior_log_prob_cuda,#TestLKJPrior#,40

Before Change

    def test_lkj_prior_log_prob_cuda(self):
        if torch.cuda.is_available():
            return self.test_lkj_prior_log_prob(cuda=True)

    def test_lkj_prior_batch_log_prob(self, cuda=False):
        device = torch.device("cuda") if cuda else torch.device("cpu")
        prior = LKJPrior(2, torch.tensor([0.5, 1.5], device=device))

After Change

    def test_lkj_prior_log_prob_cuda(self):
        if torch.cuda.is_available():
            with least_used_cuda_device():
                self.test_lkj_prior_log_prob(cuda=True)

    def test_lkj_prior_batch_log_prob(self, cuda=False):
        device = torch.device("cuda") if cuda else torch.device("cpu")
        prior = LKJPrior(2, torch.tensor([0.5, 1.5], device=device))

Instances