code clean

fleetrec/models/ctr_dnn/data_generator.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from __future__ import print_function
-from abc import ABC
 
 from fleetrec.reader.reader import Reader
 from fleetrec.utils import envs
@@ -59,7 +58,3 @@ class TrainReader(Reader):
             yield zip(feature_name, [dense_feature] + sparse_feature + [label])
 
         return reader
-
-
-class EvaluateReader(Reader, ABC):
-    pass

fleetrec/trainer/transpiler_trainer.py

@@ -43,7 +43,7 @@ class TranspileTrainer(Trainer):
         batch_size = envs.get_global_env("batch_size", None, namespace)
         reader_class = envs.get_global_env("class", None, namespace)
         abs_dir = os.path.dirname(os.path.abspath(__file__))
-        reader = os.path.join(abs_dir, '..', 'reader_implement.py')
+        reader = os.path.join(abs_dir, '../reader', 'reader_instance.py')
         pipe_cmd = "python {} {} {} {}".format(reader, reader_class, "TRAIN", self._config)
         train_data_path = envs.get_global_env("train_data_path", None, namespace)