Fix missing keras module when auto-generating docs

Lifann · rhdong · commit ed04c3cb86a1 · 2022-03-29T13:26:43.000+08:00
diff --git a/tensorflow_recommenders_addons/dynamic_embedding/__init__.py b/tensorflow_recommenders_addons/dynamic_embedding/__init__.py
@@ -37,6 +37,7 @@
     'enable_inference_mode',
     'enable_train_mode',
     'get_model_mode',
+    'keras',
     'math',
     'shadow_ops',
 ]
