Commit 73d48524 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Fixes usage of tf.app in official model repos.

PiperOrigin-RevId: 308369201
parent 652cf785
......@@ -31,7 +31,7 @@ import numpy as np
import pandas as pd
import six
from six.moves import urllib # pylint: disable=redefined-builtin
from absl import app as absl_app
from absl import app
from absl import flags
from absl import logging
import tensorflow as tf
......@@ -40,7 +40,6 @@ import tensorflow as tf
from official.utils.flags import core as flags_core
ML_1M = "ml-1m"
ML_20M = "ml-20m"
DATASETS = [ML_1M, ML_20M]
......@@ -306,4 +305,4 @@ def main(_):
if __name__ == "__main__":
define_data_download_flags()
FLAGS = flags.FLAGS
absl_app.run(main)
app.run(main)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment