From 1915c35964768c8eab7e46531d9d44bb62694e20 Mon Sep 17 00:00:00 2001
From: Ryan Peach <rgpeach10@gmail.com>
Date: Wed, 27 Mar 2024 22:02:32 -0400
Subject: [PATCH] Now checks the build file

---
 scripts/py_typed.sh | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/scripts/py_typed.sh b/scripts/py_typed.sh
index 31ac19cef97eb..c0be225678426 100755
--- a/scripts/py_typed.sh
+++ b/scripts/py_typed.sh
@@ -46,6 +46,12 @@ process_directory() {
                 echo "ERROR: Missing py.typed in $dir"
                 MISSING_PY_TYPED=1 # Mark as missing
             fi
+
+            # Check that BUILD file is present and contains the expected content
+            if [[ ! -f "$dir/BUILD" || ! $(grep -q "py_typed" "$dir/BUILD") ]]; then
+                echo "ERROR: Missing or incorrect BUILD file in $dir"
+                MISSING_PY_TYPED=1 # Mark as missing
+            fi
         fi
     fi