File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -689,12 +689,13 @@ def test_dynamically_switch_inference_training_mode(self) -> None:
689689 self .assertTrue (m ._eviction_policy_name is None )
690690 self .assertTrue (m ._eviction_module is None )
691691
692+ # Skipping this test because it is flaky on CI. TODO: T240185573 T240185565 investigate the flakiness and re-enable the test.
692693 # Pyre-ignore [56]: Pyre was not able to infer the type of argument `torch.cuda.device_count() < 1` to decorator factory `unittest.skipIf`
693694 @unittest .skipIf (
694695 torch .cuda .device_count () < 1 ,
695696 "Not enough GPUs, this test requires at least two GPUs" ,
696697 )
697- def test_zch_hash_disable_fallback (self ) -> None :
698+ def test_zch_hash_disable_fallback_disabled_in_oss_compatatibility (self ) -> None :
698699 m = HashZchManagedCollisionModule (
699700 zch_size = 30 ,
700701 device = torch .device ("cuda" ),
You can’t perform that action at this time.
0 commit comments