[PromptFlow] Optimize memory computation #4363
Workflow file for this run
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: promptflow-executor-unit-test | |
| on: | |
| schedule: | |
| - cron: "40 19 * * *" # Every day starting at 3:40 BJT | |
| pull_request_target: | |
| paths: | |
| - src/promptflow/* | |
| - src/promptflow/promptflow/* | |
| - src/promptflow/promptflow/_core/** | |
| - src/promptflow/promptflow/_internal/** | |
| - src/promptflow/promptflow/_utils/** | |
| - src/promptflow/promptflow/batch/** | |
| - src/promptflow/promptflow/contracts/** | |
| - src/promptflow/promptflow/entities/** | |
| - src/promptflow/promptflow/executor/** | |
| - src/promptflow/promptflow/integrations/** | |
| - src/promptflow/promptflow/storage/** | |
| - src/promptflow/tests/* | |
| - src/promptflow/tests/executor/** | |
| - scripts/building/** | |
| - .github/workflows/promptflow-executor-unit-test.yml | |
| workflow_dispatch: | |
| env: | |
| packageSetupType: promptflow_with_extra | |
| testWorkingDirectory: ${{ github.workspace }}/src/promptflow | |
| PYTHONPATH: ${{ github.workspace }}/src/promptflow | |
| IS_IN_CI_PIPELINE: "true" | |
| jobs: | |
| authorize: | |
| environment: | |
| # forked prs from pull_request_target will be run in external environment, domain prs will be run in internal environment | |
| ${{ github.event_name == 'pull_request_target' && | |
| github.event.pull_request.head.repo.full_name != github.repository && | |
| 'external' || 'internal' }} | |
| runs-on: ubuntu-latest | |
| steps: | |
| - run: true | |
| build: | |
| needs: authorize | |
| strategy: | |
| fail-fast: false | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: checkout | |
| uses: actions/checkout@v4 | |
| with: | |
| ref: ${{ github.event.pull_request.head.sha || github.ref }} | |
| fetch-depth: 0 | |
| - name: merge main to current branch | |
| uses: "./.github/actions/step_merge_main" | |
| - name: Display and Set Environment Variables | |
| run: | | |
| env | sort >> $GITHUB_OUTPUT | |
| id: display_env | |
| shell: bash -el {0} | |
| - name: Python Setup - ubuntu-latest - Python Version 3.9 | |
| uses: "./.github/actions/step_create_python_environment" | |
| with: | |
| pythonVersion: 3.9 | |
| - name: Build wheel | |
| uses: "./.github/actions/step_sdk_setup" | |
| with: | |
| setupType: promptflow_with_extra | |
| scriptPath: ${{ env.testWorkingDirectory }} | |
| - name: Upload Wheel | |
| if: always() | |
| uses: actions/upload-artifact@v3 | |
| with: | |
| name: wheel | |
| path: | | |
| ${{ github.workspace }}/src/promptflow/dist/*.whl | |
| ${{ github.workspace }}/src/promptflow-tools/dist/*.whl | |
| executor_unit_tests: | |
| needs: build | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| os: [ubuntu-latest] | |
| runs-on: ${{ matrix.os }} | |
| steps: | |
| - name: checkout | |
| uses: actions/checkout@v4 | |
| with: | |
| ref: ${{ github.event.pull_request.head.sha || github.ref }} | |
| fetch-depth: 0 | |
| - name: merge main to current branch | |
| uses: "./.github/actions/step_merge_main" | |
| - name: Display and Set Environment Variables | |
| run: | | |
| env | sort >> $GITHUB_OUTPUT | |
| id: display_env | |
| shell: bash -el {0} | |
| - name: Python Setup - ${{ matrix.os }} - Python Version 3.9 | |
| uses: "./.github/actions/step_create_python_environment" | |
| with: | |
| pythonVersion: 3.9 | |
| - name: Download Artifacts | |
| uses: actions/download-artifact@v3 | |
| with: | |
| name: wheel | |
| path: artifacts | |
| - name: Install wheel | |
| shell: pwsh | |
| working-directory: artifacts | |
| run: | | |
| Set-PSDebug -Trace 1 | |
| pip install -r ${{ github.workspace }}/src/promptflow/dev_requirements.txt | |
| gci ./promptflow -Recurse | % {if ($_.Name.Contains('.whl')) {python -m pip install "$($_.FullName)"}} | |
| gci ./promptflow-tools -Recurse | % {if ($_.Name.Contains('.whl')) {python -m pip install $_.FullName}} | |
| pip freeze | |
| - name: Azure Login | |
| uses: azure/login@v1 | |
| with: | |
| creds: ${{ secrets.AZURE_CREDENTIALS }} | |
| - name: Generate Configs | |
| uses: "./.github/actions/step_generate_configs" | |
| with: | |
| targetFolder: ${{ env.testWorkingDirectory }} | |
| - name: Get number of CPU cores | |
| uses: SimenB/github-actions-cpu-cores@v1 | |
| id: cpu-cores | |
| - name: Run Coverage Test | |
| shell: pwsh | |
| working-directory: ${{ github.workspace }} | |
| run: | | |
| gci env:* | sort-object name | |
| az account show | |
| pip install langchain | |
| python scripts/building/run_coverage_tests.py ` | |
| -p ${{ env.testWorkingDirectory }}/promptflow ` | |
| -t ${{ env.testWorkingDirectory }}/tests/executor/unittests ` | |
| -l eastus ` | |
| -m "all" ` | |
| -n ${{ steps.cpu-cores.outputs.count }} ` | |
| --coverage-config ${{ env.testWorkingDirectory }}/tests/executor/.coveragerc ` | |
| --disable-cov-branch | |
| - name: Upload Test Results | |
| if: always() | |
| uses: actions/upload-artifact@v3 | |
| with: | |
| name: Test Results (Python 3.9) (OS ${{ matrix.os }}) | |
| path: | | |
| ${{ github.workspace }}/*.xml | |
| ${{ github.workspace }}/htmlcov/ | |
| publish-test-results: | |
| name: "Publish Tests Results" | |
| needs: executor_unit_tests | |
| runs-on: ubuntu-latest | |
| permissions: write-all | |
| if: always() | |
| steps: | |
| - name: checkout | |
| uses: actions/checkout@v4 | |
| with: | |
| ref: ${{ github.event.pull_request.head.sha || github.ref }} | |
| fetch-depth: 0 | |
| - name: merge main to current branch | |
| uses: "./.github/actions/step_merge_main" | |
| - name: Publish Test Results | |
| uses: "./.github/actions/step_publish_test_results" | |
| with: | |
| testActionFileName: promptflow-executor-unit-test.yml | |
| testResultTitle: Executor Unit Test Result | |
| osVersion: ubuntu-latest | |
| pythonVersion: 3.9 | |
| coverageThreshold: 50 | |
| token: ${{ secrets.GITHUB_TOKEN }} | |
| context: test/executor_unit |