Try testing in OSS

Author: mergennachin

codegen/tools/targets.bzl

@@ -196,27 +196,27 @@ def define_common_targets(is_fbcode = False):
         _is_external_target = True,
     )
 
-    if not runtime.is_oss:
-        runtime.cxx_python_extension(
-            name = "selective_build",
-            srcs = [
-                "selective_build.cpp",
-            ],
-            base_module = "executorch.codegen.tools",
-            types = ["selective_build.pyi"],
-            preprocessor_flags = [
-                "-DEXECUTORCH_PYTHON_MODULE_NAME=selective_build",
-            ],
-            deps = [
-                "//executorch/runtime/core:core",
-                "//executorch/schema:program",
-            ],
-            external_deps = [
-                "pybind11",
-            ],
-            use_static_deps = True,
-            visibility = ["//executorch/codegen/..."],
-        )
+    
+    runtime.cxx_python_extension(
+        name = "selective_build",
+        srcs = [
+            "selective_build.cpp",
+        ],
+        base_module = "executorch.codegen.tools",
+        types = ["selective_build.pyi"],
+        preprocessor_flags = [
+            "-DEXECUTORCH_PYTHON_MODULE_NAME=selective_build",
+        ],
+        deps = [
+            "//executorch/runtime/core:core",
+            "//executorch/schema:program",
+        ],
+        external_deps = [
+            "pybind11",
+        ],
+        use_static_deps = True,
+        visibility = ["//executorch/codegen/..."],
+    )
 
 
     # TODO(larryliu0820): This is a hack to only run these two on fbcode. These targets depends on exir which is only available in fbcode.
@@ -255,20 +255,20 @@ def define_common_targets(is_fbcode = False):
             ],
         )
 
-        runtime.python_test(
-            name = "test_selective_build",
-            srcs = [
-                "test/test_selective_build.py",
-            ],
-            package_style = "inplace",
-            visibility = [
-                "PUBLIC",
-            ],
-            deps = [
-                ":selective_build",
-                "fbsource//third-party/pypi/expecttest:expecttest",
-                "//caffe2:torch",
-                "//executorch/exir:lib",
-            ],
-            _is_external_target = True,
-        )
+    runtime.python_test(
+        name = "test_selective_build",
+        srcs = [
+            "test/test_selective_build.py",
+        ],
+        package_style = "inplace",
+        visibility = [
+            "PUBLIC",
+        ],
+        deps = [
+            ":selective_build",
+            "fbsource//third-party/pypi/expecttest:expecttest",
+            "//caffe2:torch",
+            "//executorch/exir:lib",
+        ],
+        _is_external_target = True,
+    )

extension/pytree/TARGETS

@@ -15,9 +15,11 @@ runtime.cxx_python_extension(
     ],
     base_module = "executorch.extension.pytree",
     deps = [
-        "fbsource//third-party/pybind11:pybind11",
         ":pytree",
     ],
+    external_deps = [
+        "pybind11",
+    ],
 )
 
 runtime.cxx_python_extension(
@@ -27,9 +29,11 @@ runtime.cxx_python_extension(
     ],
     base_module = "executorch.extension.pytree",
     deps = [
-        "fbsource//third-party/pybind11:pybind11",
         ":pytree",
     ],
+    external_deps = [
+        "pybind11",
+    ],
 )
 
 runtime.python_library(

pytest.ini

@@ -31,6 +31,7 @@ addopts =
 
     # codegen
     codegen/test
+    codegen/tools
 
     # devtools
     devtools/

shim_et/xplat/executorch/build/runtime_wrapper.bzl

@@ -346,6 +346,7 @@ def _python_binary(*args, **kwargs):
 
 def _python_test(*args, **kwargs):
     _patch_kwargs_common(kwargs)
+    _remove_caffe2_deps(kwargs)
     env.python_test(*args, **kwargs)
 
 def get_oss_build_kwargs():