Skip to content

Commit cd9a5d1

Browse files
committed
update
1 parent de6a88c commit cd9a5d1

File tree

2 files changed

+629
-0
lines changed

2 files changed

+629
-0
lines changed

.github/workflows/nightly_tests.yml

Lines changed: 63 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,10 @@ env:
1515
RUN_NIGHTLY: yes
1616
PIPELINE_USAGE_CUTOFF: 5000
1717
SLACK_API_TOKEN: ${{ secrets.SLACK_CIFEEDBACK_BOT_TOKEN }}
18+
HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
19+
CONSOLIDATED_REPORT_PATH: consolidated_test_report.md
20+
SHOW_SLOWEST_TESTS: 20
21+
HF_REPORT_DATASET: diffusers-ci/nightly-test-reports
1822

1923
jobs:
2024
setup_torch_cuda_pipeline_matrix:
@@ -475,6 +479,65 @@ jobs:
475479
run: |
476480
pip install slack_sdk tabulate
477481
python utils/log_reports.py >> $GITHUB_STEP_SUMMARY
482+
483+
generate_consolidated_report:
484+
name: Generate Consolidated Test Report
485+
needs: [
486+
run_nightly_tests_for_torch_pipelines,
487+
run_nightly_tests_for_other_torch_modules,
488+
run_big_gpu_torch_tests,
489+
run_nightly_onnx_tests,
490+
run_nightly_quantization_tests,
491+
run_flax_tpu_tests
492+
]
493+
if: always()
494+
runs-on:
495+
group: aws-general-8-plus
496+
container:
497+
image: diffusers/diffusers-pytorch-cpu
498+
steps:
499+
- name: Checkout diffusers
500+
uses: actions/checkout@v3
501+
with:
502+
fetch-depth: 2
503+
504+
- name: Create reports directory
505+
run: mkdir -p combined_reports
506+
507+
- name: Download all test reports
508+
uses: actions/download-artifact@v4
509+
with:
510+
path: artifacts
511+
512+
- name: Prepare reports
513+
run: |
514+
# Move all report files to a single directory for processing
515+
find artifacts -name "*.txt" -exec cp {} combined_reports/ \;
516+
517+
- name: Install dependencies
518+
run: |
519+
pip install -e .[test]
520+
pip install slack_sdk tabulate huggingface_hub
521+
522+
- name: Generate consolidated report
523+
run: |
524+
python utils/consolidated_test_report.py \
525+
--reports_dir combined_reports \
526+
--output_file $CONSOLIDATED_REPORT_PATH \
527+
--slack_channel_name diffusers-ci-nightly \
528+
--hf_dataset_repo $HF_REPORT_DATASET \
529+
--upload_to_hub \
530+
--compare_with_previous
531+
532+
- name: Show consolidated report
533+
run: |
534+
cat $CONSOLIDATED_REPORT_PATH >> $GITHUB_STEP_SUMMARY
535+
536+
- name: Upload consolidated report
537+
uses: actions/upload-artifact@v4
538+
with:
539+
name: consolidated_test_report
540+
path: ${{ env.CONSOLIDATED_REPORT_PATH }}
478541

479542
# M1 runner currently not well supported
480543
# TODO: (Dhruv) add these back when we setup better testing for Apple Silicon

0 commit comments

Comments
 (0)