Update test_draft_export.py

pianpwk · web-flow · commit 30f51c3a3182 · 2025-02-04T16:21:29.000-08:00
diff --git a/test/export/test_draft_export.py b/test/export/test_draft_export.py
@@ -4,7 +4,7 @@
 import unittest
 
 import torch
-from torch.export import Dim, dynamic_shapes, export
+from torch.export import Dim, export
 from torch.export._draft_export import draft_export, FailureType
 from torch.testing import FileCheck
 from torch.testing._internal.common_utils import IS_WINDOWS, run_tests, TestCase
