batch prediction in case of redshift and bigquery #922
This file contains 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: Tests | |
| on: | |
| pull_request: | |
| branches: | |
| - main | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.ref }} | |
| cancel-in-progress: true | |
| jobs: | |
| unit-test: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Check out code | |
| uses: actions/checkout@v4 | |
| - name: Install Python | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: 3.10.13 | |
| cache: 'pip' | |
| - name: Install formatter | |
| run: pip install black==24.8.0 | |
| - name: Check format | |
| run: black --check . | |
| - name: Install dependencies | |
| working-directory: src/predictions | |
| run: pip3 install . | |
| - name: Compile core train | |
| run: python -m src.predictions.profiles_mlcorelib.ml_core.preprocess_and_train --mode ci | |
| - name: Compile core predict | |
| run: python -m src.predictions.profiles_mlcorelib.ml_core.preprocess_and_predict --mode ci | |
| - name: Run tests | |
| run: python -m unittest discover -s tests -p "*.py" | |
| propensity_changes: | |
| runs-on: ubuntu-latest | |
| permissions: | |
| pull-requests: read | |
| outputs: | |
| propensity: ${{ steps.filter.outputs.propensity }} | |
| steps: | |
| - uses: dorny/paths-filter@v3 | |
| id: filter | |
| with: | |
| filters: | | |
| propensity: | |
| - '!src/predictions/profiles_mlcorelib/py_native/attribution_report.py' | |
| - '!tests/integration/attribution/**' | |
| - '!samples/attribution_project/**' | |
| - '!src/predictions/profiles_mlcorelib/py_native/llm.py' | |
| - '!src/predictions/profiles_mlcorelib/py_native/id_stitcher/**' | |
| predicate-quantifier: every | |
| attribution_changes: | |
| runs-on: ubuntu-latest | |
| permissions: | |
| pull-requests: read | |
| outputs: | |
| attribution: ${{ steps.filter.outputs.attribution }} | |
| steps: | |
| - uses: dorny/paths-filter@v3 | |
| id: filter | |
| with: | |
| filters: | | |
| attribution: | |
| - 'src/predictions/setup.py' | |
| - 'samples/attribution_project/**' | |
| - '.github/**' | |
| - 'tests/integration/attribution/**' | |
| - 'tests/integration/utils.py' | |
| - 'src/predictions/profiles_mlcorelib/py_native/attribution_report.py' | |
| audit_changes: | |
| runs-on: ubuntu-latest | |
| permissions: | |
| pull-requests: read | |
| outputs: | |
| audit: ${{ steps.filter.outputs.audit }} | |
| steps: | |
| - uses: dorny/paths-filter@v3 | |
| id: filter | |
| with: | |
| filters: | | |
| audit: | |
| - 'src/predictions/setup.py' | |
| - 'samples/integration_test_project/**' | |
| - '.github/**' | |
| - 'tests/integration/audit/**' | |
| - 'tests/integration/utils.py' | |
| - 'src/predictions/profiles_mlcorelib/py_native/id_stitcher/**' | |
| propensity-integration-test: | |
| runs-on: ubuntu-latest | |
| needs: | |
| - propensity_changes | |
| - unit-test | |
| if: ${{ needs.propensity_changes.outputs.propensity == 'true' }} | |
| strategy: | |
| matrix: | |
| warehouse: | |
| - test: tests/integration/bigquery/pynative.py | |
| site_config: BIGQUERY_SITE_CONFIG | |
| python_version: 3.10.13 | |
| - test: tests/integration/snowflake/pynative.py | |
| site_config: SNOWFLAKE_SITE_CONFIG | |
| python_version: 3.10.13 | |
| - test: tests/integration/snowflake/pynative.py | |
| site_config: SNOWFLAKE_SITE_CONFIG | |
| python_version: 3.9 | |
| - test: tests/integration/redshift/pynative.py | |
| site_config: REDSHIFT_SITE_CONFIG_V2 | |
| python_version: 3.10.13 | |
| steps: | |
| - name: Check out code | |
| uses: actions/checkout@v4 | |
| - name: Install Python | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: ${{matrix.warehouse.python_version}} | |
| cache: 'pip' | |
| - name: Install local mlcorelib | |
| working-directory: src/predictions | |
| run: pip3 install . | |
| - name: Run pynative integration test | |
| run: python -m unittest ${{matrix.warehouse.test}} | |
| env: | |
| SITE_CONFIG: ${{secrets[matrix.warehouse.site_config]}} | |
| attribution-integration-test: | |
| runs-on: ubuntu-latest | |
| needs: | |
| - attribution_changes | |
| - unit-test | |
| if: ${{ needs.attribution_changes.outputs.attribution == 'true' }} | |
| strategy: | |
| matrix: | |
| warehouse: | |
| - test: tests/integration/attribution/attribution.py | |
| site_config: SNOWFLAKE_SITE_CONFIG | |
| - test: tests/integration/attribution/attribution.py | |
| site_config: REDSHIFT_SITE_CONFIG_V2 | |
| steps: | |
| - name: Check out code | |
| uses: actions/checkout@v4 | |
| - name: Install Python | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: 3.10.13 | |
| cache: 'pip' | |
| - name: Install local mlcorelib | |
| working-directory: src/predictions | |
| run: pip3 install . | |
| - name: Run pynative integration test | |
| run: python -m unittest ${{matrix.warehouse.test}} | |
| env: | |
| SITE_CONFIG: ${{secrets[matrix.warehouse.site_config]}} | |
| audit-integration-test: | |
| runs-on: ubuntu-latest | |
| needs: | |
| - audit_changes | |
| - unit-test | |
| if: ${{ needs.audit_changes.outputs.audit == 'true' }} | |
| strategy: | |
| matrix: | |
| warehouse: | |
| - test: tests/integration/audit/audit.py | |
| site_config: SNOWFLAKE_SITE_CONFIG | |
| # TODO: uncomment after integration of RS and BQ PR merges to pocs for audit | |
| # - test: tests/integration/audit/audit.py | |
| # site_config: REDSHIFT_SITE_CONFIG_V2 | |
| # - test: tests/integration/audit/audit.py | |
| # site_config: BIGQUERY_SITE_CONFIG | |
| steps: | |
| - name: Check out code | |
| uses: actions/checkout@v4 | |
| - name: Install Python | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: 3.10.13 | |
| cache: 'pip' | |
| - name: Install local mlcorelib | |
| working-directory: src/predictions | |
| run: pip3 install . | |
| - name: Run pynative integration test | |
| run: python -m unittest ${{matrix.warehouse.test}} | |
| env: | |
| SITE_CONFIG: ${{secrets[matrix.warehouse.site_config]}} |