diff --git a/scripts/update_setup_py.sh b/scripts/update_setup_py.sh
index 9f784cae..1e950cf6 100755
--- a/scripts/update_setup_py.sh
+++ b/scripts/update_setup_py.sh
@@ -43,7 +43,7 @@ echo -e "rules:
         constraint_files = set()
 
         # groups \"my-package-name<=x.y.z,...\" into (\"my-package-name\", \"<=x.y.z,...\")
-        requirement_line_regex = re.compile(r\"([a-zA-Z0-9-_.]+)([<>=][^#\s]+)?\")
+        requirement_line_regex = re.compile(r\"([a-zA-Z0-9-_.\[\]]+)([<>=][^#\s]+)?\")
 
         def add_version_constraint_or_raise(current_line, current_requirements, add_if_not_present):
             regex_match = requirement_line_regex.match(current_line)