@@ -26,15 +26,10 @@ http_archive(
 url = "https://.com/bazelbuild/rules_java/releases/download/5.3.5/rules_java-5.3.5.tar.gz",
 )
 
-# TensorFlow - using http_archive instead of git submodule
-# Commit: 49908c1e646b87987251a9bd70c7be5b9a2f2649 (2025-06-03)
-http_archive(
+# TensorFlow
+local_repository(
 name = "org_tensorflow",
-sha256 = "12ee9f1b6fdbfffc3eaf778fda7f8436cf5b853fd458ae7416f58763ddddb991",
-strip_prefix = "tensorflow-49908c1e646b87987251a9bd70c7be5b9a2f2649",
-urls = [
-"https://.com/tensorflow/tensorflow/archive/49908c1e646b87987251a9bd70c7be5b9a2f2649.tar.gz",
-],
+path = "third_party/tensorflow",
 )
 
 # Initialize the TensorFlow repository and all dependencies.

@@ -26,6 +26,7 @@ BUILD_FLAGS=(
 "--config=disable_tf_lite_py"
 "--test_lang_filters=${TEST_LANG_FILTERS}"
 "--keep_going"
+"--repo_env=USE_PYWRAP_RULES=True"
 )
 
 # Add Bazel --config flags based on kokoro injected env ie. --config=public_cache
@@ -71,20 +72,11 @@ EXCLUDED_TARGETS=(
 "-//tflite/java/..."
 "-//tflite/tools/benchmark/experimental/..."
 "-//tflite/delegates/gpu/..."
-# TODO: (b/410925271) - Remove once the test is fixed.
-"-//tflite/core/experimental/acceleration/mini_benchmark/c:c_api_test"
-"-//tflite/experimental/acceleration/..."
-"-//tflite/python:analyzer_test"
-"-//tflite/python:convert_saved_model_test"
-"-//tflite/python:convert_test"
-"-//tflite/python:test_util_test"
-"-//tflite/python/metrics:metrics_test"
-"-//tflite/toco/logging:gen_html_test"
-"-//tflite/tools:flatbuffer_utils_test"
-"-//tflite/tools:visualize_test"
+# TODO: (b/410925271) - Targets not migrated to pywrap_rules yet
 "-//tflite/tools/optimize/python:modify_model_interface_lib_test"
-"-//tflite/python/kernel_tests/signal:window_ops_test_cpu"
+"-//tflite/core/experimental/acceleration/mini_benchmark/c:c_api_test"
 "-//tflite/testing/..."
+"-//tflite/toco/..."
 )
 
 LITERT_EXCLUDED_TARGETS=(