Cleanup script tests passing

Author: evertlammerts

.github/workflows/cleanup_pypi.yml

@@ -51,10 +51,9 @@ jobs:
       - name: Run Cleanup
         run: |
           set -x
-          pypi_index_flag=${{ inputs.environment == 'production.pypi' && '--prod' || '--test' }}
           uv sync --only-group pypi --no-install-project
-          uv run --no-sync -s scripts/pypi_cleanup.py ${{ inputs.dry-run && '--dry' || '' }} \
-            ${pypi_index_flag} \
+          uv run --no-sync python -u -m duckdb_packaging.pypi_cleanup ${{ inputs.dry-run && '--dry' || '' }} \
+            ${{ inputs.environment == 'production.pypi' && '--prod' || '--test' }} \
             --username "${{ vars.PYPI_CLEANUP_USERNAME }}" \
             --max-nightlies ${{ vars.PYPI_MAX_NIGHTLIES }} > cleanup_output 2>&1
 

pyproject.toml

@@ -235,6 +235,7 @@ test = [ # dependencies used for running tests
     "polars",
     "psutil",
     "py4j",
+    "pyotp",
     "pyspark",
     "pytz",
     "requests",