remove type check

mosaicml · Sep 20, 2024 · 0dac98a · 0dac98a
1 parent 2390ae6
commit 0dac98a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/composer/utils/parallelism.py b/composer/utils/parallelism.py
@@ -4,7 +4,7 @@
 """Parallelism configs."""
 
 from dataclasses import dataclass, field
-from typing import Any, Optional, Union
+from typing import Any, Optional
 
 from torch.distributed._tensor.device_mesh import DeviceMesh