Skip to content

Commit 4a30225

Browse files
authored
Fix entrypoint.sh script (#37)
Add argument checking, avoid hard coded dir name.
1 parent dd52e4f commit 4a30225

2 files changed

Lines changed: 12 additions & 2 deletions

File tree

entrypoint.sh

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,17 @@
22

33
set -eu
44

5+
if [[ $# -lt 2 ]]; then
6+
echo "Error: Expected ./entrypoint.sh <input_file> <output_dir>"
7+
exit 1
8+
fi
9+
510
echo "Extracting feature layers from $1"
611
./process.sh "$@"
712

8-
for input_file in out/*.parquet; do
13+
for input_file in "$2"/*.parquet; do
914
echo "Sorting and compressing $input_file"
10-
output_file="out/$(basename -s .parquet $input_file)-optimized.parquet"
15+
output_file="$2/$(basename -s .parquet $input_file)-optimized.parquet"
1116
./postprocess.sh $input_file $output_file
1217
mv $output_file $input_file
1318
done

process.sh

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,11 @@
1515

1616
set -eu
1717

18+
if [[ $# -lt 2 ]]; then
19+
echo "Error: Expected ./process.sh <input_file> <output_dir> ..."
20+
exit 1
21+
fi
22+
1823
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
1924

2025
INPUT="$1"

0 commit comments

Comments
 (0)