update exception message (NVIDIA#1524)

Signed-off-by: Masaki Kozuki <mkozuki@nvidia.com> Signed-off-by: Masaki Kozuki <mkozuki@nvidia.com>
yuanzhedong · Jul 14, 2023 · b08e6c1 · b08e6c1
1 parent 5a452c0
commit b08e6c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/L0/run_transformer/test_pipeline_parallel_fwd_bwd.py b/tests/L0/run_transformer/test_pipeline_parallel_fwd_bwd.py
@@ -321,7 +321,7 @@ def _test_hybrid_backends(self, forward_only: bool) -> None:
         else:
             with self.assertRaisesRegex(
                 ImportError,
-                re.escape("UCC backend requires [torch_ucc](https://github.com/facebookresearch/torch_ucc) but not found"),
+                re.escape("UCC backend requires pytorch source build with UCC installed and enabled"),
             ):
                 self._run_hybrid_distributed_backend(forward_only)