def disabled_testFailureCase(self): # pylint: disable=invalid-name # TODO(b/140229057): This test should pass. dist = tfd.Chi(df=np.float32(27.744131) * np.ones((4,)).astype(np.float32)) dist = tfd.TransformedDistribution( bijector=tfb.NormalCDF(), distribution=dist) dist = tfb.Expm1()(dist) samps = 1.7182817 + tf.zeros_like(dist.sample(seed=test_util.test_seed())) self.assertAllClose(dist.log_prob(samps)[0], dist[0].log_prob(samps[0]))
def disabled_testFailureCase(self): # TODO(b/140229057): This test should pass. dist = tfd.Chi(df=np.float32(27.744131)) dist = tfd.TransformedDistribution( bijector=tfb.NormalCDF(), distribution=dist, batch_shape=[4]) dist = tfb.Expm1()(dist) samps = 1.7182817 + tf.zeros_like(dist.sample(seed=test_util.test_seed())) self.assertAllClose(dist.log_prob(samps)[0], dist[0].log_prob(samps[0]))
def testComposeFromTDSubclassWithAlternateCtorArgs(self): # This line used to raise an exception. tfb.Identity()(tfd.Chi(df=1., allow_nan_stats=True))