We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 04b15e9 commit 2996a9fCopy full SHA for 2996a9f
torchrec/schema/api_tests/test_planner_schema.py
@@ -37,7 +37,7 @@
37
StorageReservation,
38
Topology,
39
)
40
-from torchrec.distributed.types import ModuleSharder, PipelineType, ShardingPlan
+from torchrec.distributed.types import ModuleSharder, PipelineType
41
from torchrec.schema.utils import is_signature_compatible
42
43
0 commit comments