chore: add back collect script (#633)

Author: christinewangcw

.github/actions/run-ats/ats.sh

@@ -50,7 +50,7 @@ if [ ! -s codecov_ats/tests_to_run.txt ]; then
     echo "No tests to run, collecting from default tests"
     PYTEST_ARGS="${COLLECT_ARGS} ${DEFAULT_TESTS}"
     echo "Using args: ${PYTEST_ARGS}"
-    TESTS_TO_RUN=$(PYTEST_ARGS=${PYTEST_ARGS} ./.circleci/collect.sh)
+    TESTS_TO_RUN=$(PYTEST_ARGS=${PYTEST_ARGS} ./.github/actions/run-ats/collect.sh)
     echo "${TESTS_TO_RUN}" > codecov_ats/tests_to_run.txt
     run_count=1
     echo "Added ${TESTS_TO_RUN} as fallback. New run count: $run_count"

.github/actions/run-ats/collect.sh

@@ -0,0 +1,4 @@
+#!/usr/bin/env bash
+TESTS_TO_RUN=$(uv run --frozen pytest --collect-only ${PYTEST_ARGS} -q --disable-warnings --no-summary --no-header)
+TESTS_TO_RUN=$(echo "${TESTS_TO_RUN}" | head -n -2)
+echo $TESTS_TO_RUN