Merge pull request #1199 from jburnim/r0.12

Prepare branch for the TFP 0.12.0 release
tensorflow · Dec 22, 2020 · dcd59ed · dcd59ed
2 parents cc2c37e + 90fa3f0
commit dcd59ed
Show file tree

Hide file tree

Showing 132 changed files with 4,545 additions and 1,332 deletions.
diff --git a/discussion/fun_mcmc/fun_mcmc_test.py b/discussion/fun_mcmc/fun_mcmc_test.py
@@ -371,8 +371,8 @@ def log_prob_fn(x, y):
               tfp.distributions.Normal(self._constant(1.), 1.).log_prob(y)), ()
 
     bijectors = [
-        tfp.bijectors.AffineScalar(scale=self._constant(2.)),
-        tfp.bijectors.AffineScalar(scale=self._constant(3.))
+        tfp.bijectors.Scale(scale=self._constant(2.)),
+        tfp.bijectors.Scale(scale=self._constant(3.))
     ]
 
     (transformed_log_prob_fn,
@@ -398,8 +398,8 @@ def log_prob_fn(x, y):
               tfp.distributions.Normal(self._constant(1.), 1.).log_prob(y)), ()
 
     bijectors = {
-        'x': tfp.bijectors.AffineScalar(scale=self._constant(2.)),
-        'y': tfp.bijectors.AffineScalar(scale=self._constant(3.))
+        'x': tfp.bijectors.Scale(scale=self._constant(2.)),
+        'y': tfp.bijectors.Scale(scale=self._constant(3.))
     }
 
     (transformed_log_prob_fn,

diff --git a/spinoffs/oryx/oryx/core/interpreters/inverse/BUILD b/spinoffs/oryx/oryx/core/interpreters/inverse/BUILD
@@ -93,6 +93,11 @@ py_test(
     name = "inverse_test",
     srcs = ["inverse_test.py"],
     python_version = "PY3",
+    # This test no longer works after cl/346850541, because no inverse is registered
+    # for convert_element_type.
+    tags = [
+        "notap",
+    ],
     deps = [
         ":core",
         ":rules",