Changes to drawing for triton puzzle #116

	# This workflow will install Python dependencies, run tests and lint with a single version of Python
	# For more information see: https://docs.github.com/en/actions/automating-builds-and-tests/building-and-testing-python

	name: Python application

	on:
	push:
	branches: [ "main" ]
	pull_request:
	branches: [ "main" ]

	permissions:
	contents: read

	jobs:
	build:
	runs-on: ubuntu-latest

	steps:
	- uses: actions/checkout@v3
	with:
	path: 'triton_viz'

	- name: Set up Python 3.10
	uses: actions/setup-python@v3
	with:
	python-version: '3.10'

	- name: Cache Dependencies
	uses: actions/cache@v3
	id: cache-pip
	with:
	path: /opt/hostedtoolcache/Python/3.10.13/x64
	key: ${{ runner.os }}-pip-3.10-${{ hashFiles('**/setup.py') }}
	restore-keys: \|
	${{ runner.os }}-pip-

	- name: Clone Triton and Install
	run: \|
	git clone https://github.com/openai/triton.git
	cd triton/python
	pip install -e .

	- name: Install Dependencies if Cache Missed
	if: steps.cache-pip.outputs.cache-hit != 'true'
	run: \|
	cd triton_viz
	pip install -e .
	pre-commit install
	pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
	pip uninstall pytorch-triton -y

	- name: Lint with pre-commit
	run: \|
	cd triton_viz
	pre-commit run --all-files

	- name: Test with pytest
	run: \|
	cd triton_viz
	python -m pytest Examples

Provide feedback