Skip some more tests with Tensor.name resolution issues.

PiperOrigin-RevId: 527958188

Author: SiegeLordEx

tensorflow_probability/python/sts/fitting_test.py

@@ -340,6 +340,7 @@ class HMCTestsStatic32(test_util.TestCase, _HMCTests):
   @test_util.disable_test_for_backend(
       disable_jax=True, reason='No variables in JAX backend.')
   def test_chain_batch_shape(self, shape_in, expected_batch_shape_out):
+    self.skipTest('b/275876892')
     batch_shape = [2, 3]
     num_results = 1
     num_timesteps = 5

tensorflow_probability/python/sts/forecast_test.py

@@ -190,6 +190,7 @@ def _run():
 
   @test_util.jax_disable_test_missing_functionality('fit_with_hmc')
   def test_forecast_from_hmc(self):
+    self.skipTest('b/275876892')
     if not tf1.control_flow_v2_enabled():
       self.skipTest('test_forecast_from_hmc does not currently work with TF1')
 
@@ -202,24 +203,13 @@ def test_forecast_from_hmc(self):
     observed_time_series = self._build_tensor(np.random.randn(
         *(batch_shape + [num_timesteps])))
     model = self._build_model(observed_time_series)
-    try:
-      samples, _ = fitting.fit_with_hmc(
-          model,
-          observed_time_series,
-          num_results=num_results,
-          num_warmup_steps=2,
-          num_variational_steps=2,
-      )
-    except NotImplementedError as e:
-      err_str = str(e)
-      if "'Tensor' object has no attribute '_name'" in err_str:
-        # TODO(b/279596122): Enable the test after the upstream issue is fixed.
-        self.skipTest(
-            'test_forecast_from_hmc is failing due to Tensor object'
-            'has no attribute `shape`.'
-        )
-      else:
-        raise e
+    samples, _ = fitting.fit_with_hmc(
+        model,
+        observed_time_series,
+        num_results=num_results,
+        num_warmup_steps=2,
+        num_variational_steps=2,
+    )
 
     forecast_dist = forecast.forecast(
         model,