diff --git a/distrax/_src/bijectors/bijector_from_tfp_test.py b/distrax/_src/bijectors/bijector_from_tfp_test.py index bedef883..ec13fd90 100644 --- a/distrax/_src/bijectors/bijector_from_tfp_test.py +++ b/distrax/_src/bijectors/bijector_from_tfp_test.py @@ -217,7 +217,7 @@ def test_works_with_tfp_caching(self, tfp_bij_name, batch_shape_in, def test_access_properties_tfp_bijector(self): tfp_bij = self._test_bijectors['BatchedChain'] bij = bijector_from_tfp.BijectorFromTFP(tfp_bij) - # Access the attribute `bijectors` + # Access the attribute `bijectors`. np.testing.assert_allclose( bij.bijectors[0].shift, tfp_bij.bijectors[0].shift, atol=1e-8) np.testing.assert_allclose(