Re-enable tests after fixes to graph dismantling.

PiperOrigin-RevId: 529136208

Author: rjpower

tensorflow_probability/python/sts/fitting_test.py

@@ -245,7 +245,6 @@ def test_basic_hmc_example(self):
   @test_util.disable_test_for_backend(
       disable_jax=True, reason='No variables in JAX backend.')
   def test_multiple_chains_example(self):
-    self.skipTest('b/275876892')
     batch_shape = [2, 3]
     num_timesteps = 5
     num_results = 6

tensorflow_probability/python/sts/forecast_test.py

@@ -190,7 +190,6 @@ 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')