Disable flaky test on OSS.
@@ -152,6 +152,7 @@ distribute_py_test(
shard_count = 4, # TODO(b/184290570): Investigate why only 1 shard times out.
tags = [
"multi_and_single_gpu",
+ "no_oss", # TODO(b/194935930): Flaky test
"nomultivm", # TODO(b/170502145)
],
deps = [
