Re-enable session_test.py
A number of CL's have split up session_test.py to be a bit smaller. As a result, this CL will re-enable the session_test to see if it remains flaky. PiperOrigin-RevId: 157786407
This commit is contained in:
parent
d741d81c5f
commit
23cdf96b85
@ -2945,38 +2945,34 @@ tf_cuda_library(
|
||||
alwayslink = 1,
|
||||
)
|
||||
|
||||
# Disabled due to http://b/62145493
|
||||
# py_test(
|
||||
# name = "session_test",
|
||||
# size = "medium", # http://b/62144199
|
||||
# srcs = ["client/session_test.py"],
|
||||
# srcs_version = "PY2AND3",
|
||||
# tags = [
|
||||
# "no_gpu",
|
||||
# "no_pip_gpu", # testInteractivePlacePrunedGraph fails on invalid assumption about GPU ops.
|
||||
# ],
|
||||
# deps = [
|
||||
# ":array_ops",
|
||||
# ":client",
|
||||
# ":construction_fails_op",
|
||||
# ":control_flow_ops",
|
||||
# ":data_flow_ops",
|
||||
# ":errors",
|
||||
# ":framework",
|
||||
# ":framework_for_generated_wrappers",
|
||||
# ":framework_test_lib",
|
||||
# ":math_ops",
|
||||
# ":platform_test",
|
||||
# ":state_ops",
|
||||
# ":training",
|
||||
# ":util",
|
||||
# ":variables",
|
||||
# "//third_party/py/numpy",
|
||||
# "@six_archive//:six",
|
||||
# "//tensorflow/core/distributed_runtime/rpc:grpc_server_lib",
|
||||
# "//tensorflow/core/distributed_runtime/rpc:grpc_session",
|
||||
# ],
|
||||
# )
|
||||
py_test(
|
||||
name = "session_test",
|
||||
size = "small",
|
||||
srcs = ["client/session_test.py"],
|
||||
srcs_version = "PY2AND3",
|
||||
tags = [
|
||||
"no_gpu",
|
||||
"no_pip_gpu", # testInteractivePlacePrunedGraph fails on invalid assumption about GPU ops.
|
||||
],
|
||||
deps = [
|
||||
":array_ops",
|
||||
":client",
|
||||
":control_flow_ops",
|
||||
":data_flow_ops",
|
||||
":errors",
|
||||
":framework",
|
||||
":framework_for_generated_wrappers",
|
||||
":framework_test_lib",
|
||||
":math_ops",
|
||||
":platform_test",
|
||||
":state_ops",
|
||||
":training",
|
||||
":util",
|
||||
":variables",
|
||||
"//third_party/py/numpy",
|
||||
"@six_archive//:six",
|
||||
],
|
||||
)
|
||||
|
||||
py_test(
|
||||
name = "session_clusterspec_prop_test",
|
||||
|
Loading…
x
Reference in New Issue
Block a user