Add option to invoke AllocateSpaceForPayload from update_device.py am: 51aad993dd am: e11e1c77c6 am: bf78189670

Original change: https://android-review.googlesource.com/c/platform/system/update_engine/+/1596612

MUST ONLY BE SUBMITTED BY AUTOMERGER

Change-Id: Iddb9fc3243c6bc69e43a7c32403a0f4fade88b00
diff --git a/scripts/brillo_update_payload b/scripts/brillo_update_payload
index c510148..5b4f959 100755
--- a/scripts/brillo_update_payload
+++ b/scripts/brillo_update_payload
@@ -89,12 +89,14 @@
   exit 1
 }
 
-# Loads shflags. We first look at the default install location; then look for
-# crosutils (chroot); finally check our own directory.
+# Loads shflags. We first look at the default install location; then our own
+# directory; finally the parent directory.
 load_shflags() {
   local my_dir="$(dirname "$(readlink -f "$0")")"
   local path
-  for path in /usr/share/misc "${my_dir}"/lib/shflags; do
+  for path in /usr/share/misc \
+    "${my_dir}"/lib/shflags \
+    "${my_dir}"/../lib/shflags; do
     if [[ -r "${path}/shflags" ]]; then
       . "${path}/shflags" || die "Could not load ${path}/shflags."
       return