Skip to content

Commit

Permalink
exspec-after.sh: use uvx to run artistools
Browse files Browse the repository at this point in the history
  • Loading branch information
lukeshingles committed Jan 16, 2025
1 parent 10d1f68 commit f3321c2
Showing 1 changed file with 4 additions and 6 deletions.
10 changes: 4 additions & 6 deletions scripts/exspec-after.sh
Original file line number Diff line number Diff line change
Expand Up @@ -40,19 +40,17 @@ if [[ -f emission.out || -f emission.out.zstd ]]; then
mkdir -p speclc_angle_res
mv *_res_*.out* speclc_angle_res/ || true

python3 -m ensurepip

python3 -m pip install artistools
curl -LsSf https://astral.sh/uv/install.sh | sh

# convert packets to parquet for fast reading
python3 -m artistools lc --frompackets || true
uvx artistools lc --frompackets || true

if [ -f vpkt.txt ]; then
# convert virtual packets to parquet
python3 -m artistools lc --frompackets -plotvspecpol 0 || true
uvx artistools lc --frompackets -plotvspecpol 0 || true
fi

# convert estimators to parquet. On JUWELS, you might need to limit the number of processes to 16 in artistools/artistools/configuration.py
python3 -c 'import artistools as at; at.estimators.scan_estimators()' || true
uvx --from artistools python3 -c 'import artistools as at; at.estimators.scan_estimators()' || true

fi

0 comments on commit f3321c2

Please sign in to comment.