Allow using python functions instead of operators (e.g in pre-processing pipeline) #7214
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Test Catalog Preparation | |
| on: | |
| push: | |
| branches: [ main ] | |
| pull_request: | |
| branches: [ main ] | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.event_name == 'pull_request' && github.event.pull_request.number || github.ref_name }} | |
| cancel-in-progress: true | |
| jobs: | |
| preparation: | |
| runs-on: ubuntu-latest | |
| timeout-minutes: 30 | |
| env: | |
| OS: ubuntu-latest | |
| UNITXT_DEFAULT_VERBOSITY: error | |
| DATASETS_VERBOSITY: error | |
| HF_HUB_VERBOSITY: error | |
| HF_DATASETS_DISABLE_PROGRESS_BARS: "True" | |
| HF_HUB_DOWNLOAD_TIMEOUT: 60 | |
| HF_HUB_ETAG_TIMEOUT: 60 | |
| TQDM_DISABLE: "True" | |
| strategy: | |
| matrix: | |
| modulo: [0,1,2,3,4,5,6,7] | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - uses: actions/setup-python@v5 | |
| with: | |
| python-version: '3.9' | |
| cache: 'pip' | |
| - run: echo "blis==0" > constraints.txt | |
| - run: curl -LsSf https://astral.sh/uv/install.sh | sh | |
| - run: uv pip install --upgrade --system torch --index-url https://download.pytorch.org/whl/cpu | |
| - run: uv pip install --system -c constraints.txt -e ".[tests]" | |
| - run: | | |
| pip install --only-binary :all: spacy | |
| pip install networkx==3.2.1 | |
| pip install datasets==3.5.1 | |
| - name: Hugging Face Login | |
| run: | | |
| for i in {1..5}; do | |
| huggingface-cli login --token ${{ secrets.UNITXT_READ_HUGGINGFACE_HUB_FOR_TESTS }} && break || sleep $((2 ** i)); | |
| done | |
| - name: Run Tests | |
| run: | | |
| modulo="${{ matrix.modulo }}" | |
| echo "modulo=${modulo}" >> $GITHUB_STEP_SUMMARY | |
| echo "sed -i 's/^num_par = 1 /num_par = 8 /' tests/catalog/test_preparation.py" > sedit.sh | |
| echo "sed -i 's/^modulo = 0/modulo = ${modulo}/' tests/catalog/test_preparation.py" >> sedit.sh | |
| sh sedit.sh | |
| python -m unittest tests.catalog.test_preparation | |