MixtureSameFamilyFixed ->MixtureSameFamily

pytorch · Feb 2, 2024 · 6cff293 · 6cff293
1 parent 51782e0
commit 6cff293
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/distributions/test_distributions.py b/test/distributions/test_distributions.py
@@ -4051,7 +4051,7 @@ def test_continuous_bernoulli_shape_tensor_params(self):
     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]))
-        gmm = MixtureSameFamilyFixed(mix_distribution, component_distribution)
+        gmm = MixtureSameFamily(mix_distribution, component_distribution)
         self.assertEqual( len(gmm.mean.shape), 2)
 
 @skipIfTorchDynamo("Not a TorchDynamo suitable test")