Disable the failing test in OSS for now.
@@ -733,6 +733,7 @@ distribute_py_test(
shard_count = 2,
tags = [
"multi_and_single_gpu",
+ "no_oss", # TODO(jmullenbach): Broken by cl/439357690
"no_rocm",
"nomultivm", # TODO(b/170502145)
],
