polish code

hpcaitech · Oct 18, 2023 · b94b09f · b94b09f
1 parent 745a91a
commit b94b09f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/colossalai/tensor/d_tensor/layout_converter.py b/colossalai/tensor/d_tensor/layout_converter.py
@@ -446,7 +446,7 @@ def layout_converting(
             # Cache hit
 
             def _group_alive_check(cached_comm_action_sequence):
-                """
+                r"""
                 Check if the process groups required for sharding have been deleted by torch.distributed.destroy_process_group method.
                 If not deleted, return True; otherwise, return False.