diff --git a/TensorFlow/built-in/recommendation/WideDeep_ID2940_for_TensorFlow/trainer/task.py b/TensorFlow/built-in/recommendation/WideDeep_ID2940_for_TensorFlow/trainer/task.py
index dc8bc7bbc2038d522b28f696bc560f80135859ce..cabfd224edb430f24125724d45c1e15e3032400f 100644
--- a/TensorFlow/built-in/recommendation/WideDeep_ID2940_for_TensorFlow/trainer/task.py
+++ b/TensorFlow/built-in/recommendation/WideDeep_ID2940_for_TensorFlow/trainer/task.py
@@ -24,6 +24,8 @@ import json
 import numpy as np
 import os
 import tensorflow as tf
+tf.enable_control_flow_v2()
+tf.enable_resource_variables()
 import tensorflow_transform as tft
 from tensorflow.core.protobuf import rewriter_config_pb2
 from trainer import features