use constant instead of string literal

tensorflow · Sep 30, 2022 · 61eb5e4 · 61eb5e4
1 parent 1e85aa8
commit 61eb5e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_recommenders_addons/dynamic_embedding/python/ops/shadow_embedding_ops.py b/tensorflow_recommenders_addons/dynamic_embedding/python/ops/shadow_embedding_ops.py
@@ -247,7 +247,7 @@ def embedding_lookup(
                      ' {},'.format(ids.dtype, shadow.ids.dtype))
 
   with ops.name_scope(name, "shadow_embedding_lookup"):
-    if de.ModelMode.CURRENT_SETTING == 'train':
+    if de.ModelMode.CURRENT_SETTING == de.ModelMode.TRAIN:
       with ops.control_dependencies([shadow._reset_ids(ids)]):
         return shadow.read_value(do_prefetch=True)
     else: