[Ez][BE]: Remove accidental classvar (#153540)

Skylion007 · pytorchmergebot · commit e0dece510b70 · 2025-05-14T21:55:56.000Z
Untyped variables become ClassVar in dataclasses, this type alias should just be a type alias; no need for it to eb a classvar. Pull Request resolved: #153540 Approved by: https://github.com/albanD, https://github.com/aorenste
diff --git a/torch/distributed/_shard/sharding_spec/chunk_sharding_spec.py b/torch/distributed/_shard/sharding_spec/chunk_sharding_spec.py
@@ -1,6 +1,7 @@
 # mypy: allow-untyped-defs
 from dataclasses import dataclass
 from typing import cast, Optional, TYPE_CHECKING, Union
+from typing_extensions import TypeAlias
 
 import torch
 import torch.distributed as dist
@@ -22,6 +23,8 @@
     # from run-time to resolve circular dependency.
     from torch.distributed._shard.sharded_tensor import ShardedTensor
 
+_ShardingDim: TypeAlias = Union[int, str]
+
 
 @dataclass
 class ChunkShardingSpec(ShardingSpec):
@@ -50,9 +53,7 @@ class ChunkShardingSpec(ShardingSpec):
             :class:`torch.distributed._remote_device`
     """
 
-    ShardingDim = Union[int, str]
-
-    dim: ShardingDim
+    dim: _ShardingDim
     placements: list[Union[torch.distributed._remote_device, str]]
 
     def __post_init__(self):
