Skip to content

Commit e56e9db

Browse files
Yixin Baofacebook-github-bot
authored andcommitted
Dsiable flaky test in oss test_zch_hash_disable_fallback (#3425)
Summary: as titled. Differential Revision: D83786896
1 parent d958b4d commit e56e9db

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torchrec/modules/tests/test_hash_mc_modules.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -694,7 +694,8 @@ def test_dynamically_switch_inference_training_mode(self) -> None:
694694
torch.cuda.device_count() < 1,
695695
"Not enough GPUs, this test requires at least two GPUs",
696696
)
697-
def test_zch_hash_disable_fallback(self) -> None:
697+
def test_zch_hash_disable_fallback_disabled_in_oss_compatibility(self) -> None:
698+
# TODO: fix this flaky test in OSS.
698699
m = HashZchManagedCollisionModule(
699700
zch_size=30,
700701
device=torch.device("cuda"),

0 commit comments

Comments
 (0)