message: Fix tests in OSS keras which failed in graph mode.
diff: @@ -74,6 +74,7 @@ class RandomFourierFeaturesTest(test.TestCase, parameterized.TestCase):
  @testing_utils.run_v2_only
  def test_state_saving_and_loading(self):
  + with self.cached_session():
  input_data = np.random.random((1, 2))
  rff_layer = kernel_layers.RandomFourierFeatures(output_dim=10, scale=3.0)
  inputs = input_layer.Input((2,))
  