spaces after commas

As caught by CI
pytorch · Feb 2, 2024 · f01b7f4 · f01b7f4
1 parent e8c2f06
commit f01b7f4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/distributions/test_distributions.py b/test/distributions/test_distributions.py
@@ -4049,8 +4049,8 @@ def test_continuous_bernoulli_shape_tensor_params(self):
         self.assertEqual(continuous_bernoulli.log_prob(torch.ones(3, 1, 1)).size(), torch.Size((3, 3, 2)))
 
     def test_mixture_same_family_mean_shape(self):
-        mix_distribution = Categorical(torch.ones([3,1,3]))
-        component_distribution = Normal(torch.zeros([3,3,3]), torch.ones([3,3,3]))
+        mix_distribution = Categorical(torch.ones([3, 1, 3]))
+        component_distribution = Normal(torch.zeros([3, 3, 3]), torch.ones([3, 3, 3]))
         gmm = MixtureSameFamily(mix_distribution, component_distribution)
         self.assertEqual(len(gmm.mean.shape), 2)