Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
official/recommendation/movielens.py
View file @ 73d48524
...@@ -31,7 +31,7 @@ import numpy as np ...@@ -31,7 +31,7 @@ import numpy as np
import pandas as pd import pandas as pd
import six import six
from six.moves import urllib # pylint: disable=redefined-builtin 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 flags
from absl import logging from absl import logging
import tensorflow as tf import tensorflow as tf
...@@ -40,7 +40,6 @@ import tensorflow as tf ...@@ -40,7 +40,6 @@ import tensorflow as tf
from official.utils.flags import core as flags_core from official.utils.flags import core as flags_core
ML_1M = "ml-1m" ML_1M = "ml-1m"
ML_20M = "ml-20m" ML_20M = "ml-20m"
DATASETS = [ML_1M, ML_20M] DATASETS = [ML_1M, ML_20M]
...@@ -306,4 +305,4 @@ def main(_): ...@@ -306,4 +305,4 @@ def main(_):
if __name__ == "__main__": if __name__ == "__main__":
define_data_download_flags() define_data_download_flags()
FLAGS = flags.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