Fix fairness-indicators tests broken by keras and v2 compatibility changes.

PiperOrigin-RevId: 715469494

Author: vkarampudi

tensorboard_plugin/tensorboard_plugin_fairness_indicators/plugin_test.py

@@ -25,6 +25,7 @@
 from tensorboard_plugin_fairness_indicators import plugin
 from tensorboard_plugin_fairness_indicators import summary_v2
 import six
+from tensorflow import keras
 import tensorflow.compat.v1 as tf
 import tensorflow.compat.v2 as tf2
 import tensorflow_model_analysis as tfma
@@ -77,7 +78,7 @@ def tearDown(self):
 
   def _export_keras_model(self, classifier):
     temp_eval_export_dir = os.path.join(self.get_temp_dir(), "eval_export_dir")
-    classifier.compile(optimizer=tf.keras.optimizers.Adam(), loss="mse")
+    classifier.compile(optimizer=keras.optimizers.Adam(), loss="mse")
     tf.saved_model.save(classifier, temp_eval_export_dir)
     return temp_eval_export_dir