diff --git a/torch/_export/exported_program.py b/torch/_export/exported_program.py
index a7b0fd679a5cd..5d28ea3154908 100644
--- a/torch/_export/exported_program.py
+++ b/torch/_export/exported_program.py
@@ -1,11 +1,8 @@
-import dataclasses
-from typing import Optional
 import warnings
 
 
 import torch
 import torch.fx
-import torch.utils._pytree as pytree
 
 
 # TODO(ycao): This is added to avoid breaking existing code temporarily.
@@ -32,13 +29,6 @@
 ]
 
 
-# Information to maintain user calling/returning specs
-@dataclasses.dataclass
-class CallSpec:
-    in_spec: Optional[pytree.TreeSpec]
-    out_spec: Optional[pytree.TreeSpec]
-
-
 def _create_graph_module_for_export(root, graph):
     try:
         gm = torch.fx.GraphModule(root, graph)