Skip to content

Commit

Permalink
Add Flower Datasets e2e tests to CI (#2398)
Browse files Browse the repository at this point in the history
  • Loading branch information
adam-narozniak authored Dec 20, 2023
1 parent 216fdac commit df73648
Show file tree
Hide file tree
Showing 2 changed files with 53 additions and 2 deletions.
52 changes: 52 additions & 0 deletions .github/workflows/datasets-e2e.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
name: Datasets-E2E

on:
push:
branches:
- main
paths:
- "datasets/flwr_datasets/**"
pull_request:
branches:
- main
paths:
- "datasets/flwr_datasets/**"

concurrency:
group: ${{ github.workflow }}-${{ github.ref == 'refs/heads/main' && github.run_id || github.event.pull_request.number || github.ref }}
cancel-in-progress: true

env:
FLWR_TELEMETRY_ENABLED: 0

jobs:
frameworks:
runs-on: ubuntu-22.04
timeout-minutes: 10
# Using approach described here:
# https://docs.github.com/en/actions/using-jobs/using-a-matrix-for-your-jobs
strategy:
matrix:
include:
- directory: pytorch

- directory: tensorflow

- directory: scikit-learn

name: Framework / ${{matrix.directory}}

defaults:
run:
working-directory: datasets/e2e/${{ matrix.directory }}

steps:
- uses: actions/checkout@v4
- name: Bootstrap
uses: ./.github/actions/bootstrap
with:
python-version: 3.8
- name: Install dependencies
run: python -m poetry install
- name: Run tests
run: python -m unittest discover -p '*_test.py'
3 changes: 1 addition & 2 deletions datasets/e2e/tensorflow/tensorflow_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,8 +38,7 @@ class FdsToTensorFlow(unittest.TestCase):

@classmethod
def setUpClass(cls):
"""Disable progress bar to keep the log clean.
"""
"""Disable progress bar to keep the log clean."""
disable_progress_bar()

def _create_tensorflow_dataset(self, batch_size: int) -> tf.data.Dataset:
Expand Down

0 comments on commit df73648

Please sign in to comment.