Fix version.py

Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

Author: larryliu0820

pyproject.toml

@@ -84,6 +84,13 @@ Changelog = "https://github.com/pytorch/executorch/releases"
 [project.scripts]
 flatc = "executorch.data.bin:flatc"
 
+# Tell setuptools to follow the symlink: src/executorch/exir -> exir.
+# Doing this allows us to import from executorch.exir directly in
+# editable mode.
+[tool.setuptools.packages.find]
+where = ["src/"]
+include = ["executorch.exir"]
+
 # TODO(dbort): Could use py_modules to restrict the set of modules we
 # package, and package_data to restrict the set up non-python files we
 # include. See also setuptools/discovery.py for custom finders.
@@ -97,7 +104,6 @@ flatc = "executorch.data.bin:flatc"
 "executorch.examples.apple.coreml.llama" = "examples/apple/coreml/llama"
 "executorch.examples.llm_pte_finetuning" = "examples/llm_pte_finetuning"
 "executorch.examples.models" = "examples/models"
-"executorch" = "src/executorch"
 "executorch.extension" = "extension"
 "executorch.kernels.quantized" = "kernels/quantized"
 "executorch.schema" = "schema"

setup.py

@@ -575,8 +575,7 @@ def run(self):
             # In editable mode, the package directory is the original source directory
             dst_root = self.get_package_dir(".")
         else:
-            dst_root = os.path.join(self.build_lib, self.get_package_dir("executorch"))
-
+            dst_root = os.path.join(self.build_lib, "executorch")
         # Create the version file.
         Version.write_to_python_file(os.path.join(dst_root, "version.py"))