diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml index f13adf3..9cfc6bf 100644 --- a/.github/workflows/pr-gpu.yaml +++ b/.github/workflows/pr-gpu.yaml @@ -38,7 +38,7 @@ jobs: container: ${{ matrix.container }} python_version: ${{ matrix.python_version }} gpu_num: ${{ matrix.gpu_num }} - git_repo: databricks/megablocks + git_repo: eitanturok/megablocks # todo: replace with databricks/megablocks pip_deps: "[all,testing]" pytest_command: "coverage run -m pytest tests" # todo: remove tests from pytest tests when we delete all tests outside of MegaBlocks repo pytest_markers: "gpu"