Skip to content

Commit 5b85f51

Browse files
committed
Testing fix to relative imports error
1 parent 68ddae9 commit 5b85f51

File tree

1 file changed

+16
-9
lines changed

1 file changed

+16
-9
lines changed

codegen/tools/gen_oplist.py

Lines changed: 16 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -15,20 +15,27 @@
1515

1616
try:
1717
from executorch.codegen.parse import strip_et_fields
18+
from executorch.exir._serialize import _deserialize_pte_binary
19+
from executorch.exir.schema import (
20+
EValue,
21+
KernelCall,
22+
OptionalTensorList,
23+
Tensor,
24+
TensorList
25+
)
1826
except ImportError:
1927
# If we build from source, executorch.codegen is not available.
2028
# We can use relative import instead.
2129
from ..parse import strip_et_fields
30+
from ..exir._serialize import _deserialize_pte_binary
31+
from ..exir.schema import (
32+
EValue,
33+
KernelCall,
34+
OptionalTensorList,
35+
Tensor,
36+
TensorList
37+
)
2238

23-
24-
from executorch.exir._serialize import _deserialize_pte_binary
25-
from executorch.exir.schema import (
26-
EValue,
27-
KernelCall,
28-
OptionalTensorList,
29-
Tensor,
30-
TensorList,
31-
)
3239
from torchgen.gen import LineLoader, parse_native_yaml_struct
3340
from torchgen.selective_build.operator import SelectiveBuildOperator
3441
from torchgen.selective_build.selector import merge_et_kernel_metadata

0 commit comments

Comments
 (0)