chore: Import tensorflow.keras so that 'reload(tf.keras)' is guaranteed to work.

fionalang · copybara-github · commit bb8388ea882e · 2023-10-10T15:19:17.000-07:00
PiperOrigin-RevId: 572380971
diff --git a/tests/unit/vertexai/test_remote_training.py b/tests/unit/vertexai/test_remote_training.py
@@ -70,6 +70,7 @@
 from sklearn.linear_model import _logistic
 from sklearn.model_selection import train_test_split
 import tensorflow as tf
+import tensorflow.keras
 
 
 # Manually set tensorflow version for b/295580335
