Update triton.py