85 lines
1.8 KiB
Python
85 lines
1.8 KiB
Python
licenses(["notice"]) # Apache 2.0
|
|
|
|
package(default_visibility = ["//tensorflow:internal"])
|
|
|
|
load(
|
|
"//tensorflow:tensorflow.bzl",
|
|
"tf_copts", # @unused
|
|
"tf_cc_test", # @unused
|
|
)
|
|
|
|
py_binary(
|
|
name = "tf_upgrade",
|
|
srcs = ["tf_upgrade.py"],
|
|
srcs_version = "PY2AND3",
|
|
)
|
|
|
|
py_test(
|
|
name = "tf_upgrade_test",
|
|
size = "small",
|
|
srcs = ["tf_upgrade_test.py"],
|
|
srcs_version = "PY2AND3",
|
|
deps = [
|
|
"tf_upgrade",
|
|
"//tensorflow:tensorflow_py",
|
|
],
|
|
)
|
|
|
|
# Keep for reference, this test will succeed in 0.11 but fail in 1.0
|
|
# py_test(
|
|
# name = "test_file_v0_11",
|
|
# size = "small",
|
|
# srcs = ["testdata/test_file_v0_11.py"],
|
|
# srcs_version = "PY2AND3",
|
|
# deps = [
|
|
# "//tensorflow:tensorflow_py",
|
|
# ],
|
|
# )
|
|
|
|
genrule(
|
|
name = "generate_upgraded_file",
|
|
testonly = 1,
|
|
srcs = ["testdata/test_file_v0_11.py"],
|
|
outs = [
|
|
"test_file_v1_0.py",
|
|
"report.txt",
|
|
],
|
|
cmd = ("$(location tf_upgrade)" +
|
|
" --infile $(location testdata/test_file_v0_11.py)" +
|
|
" --outfile $(location test_file_v1_0.py)" +
|
|
" --reportfile $(location report.txt)"),
|
|
tools = ["tf_upgrade"],
|
|
)
|
|
|
|
py_test(
|
|
name = "test_file_v1_0",
|
|
size = "small",
|
|
srcs = ["test_file_v1_0.py"],
|
|
srcs_version = "PY2AND3",
|
|
deps = [
|
|
"//tensorflow:tensorflow_py",
|
|
],
|
|
)
|
|
|
|
exports_files(
|
|
[
|
|
"tf_upgrade.py",
|
|
"testdata/test_file_v0_11.py",
|
|
],
|
|
)
|
|
|
|
# -----------------------------------------------------------------------------
|
|
# Google-internal targets. These must be at the end for syncrepo.
|
|
|
|
filegroup(
|
|
name = "all_files",
|
|
srcs = glob(
|
|
["**/*"],
|
|
exclude = [
|
|
"**/METADATA",
|
|
"**/OWNERS",
|
|
],
|
|
),
|
|
visibility = ["//tensorflow:__subpackages__"],
|
|
)
|