Modified

Signed-off-by: noopur <[email protected]>

.github/workflows/task_runner_basic_e2e.yml

@@ -5,42 +5,42 @@ name: Task_Runner_E2E  # Please do not modify the name as it is used in the comp
 
 on:
   schedule:
-    - cron: '0 0 * * *' # Run every day at midnight
+    - cron: "0 0 * * *" # Run every day at midnight
   workflow_dispatch:
     inputs:
       num_rounds:
-        description: 'Number of rounds to train'
+        description: "Number of rounds to train"
         required: false
-        default: '5'
+        default: "5"
         type: string
       num_collaborators:
-        description: 'Number of collaborators'
+        description: "Number of collaborators"
         required: false
-        default: '2'
+        default: "2"
         type: string
       model_name:
-        description: 'Model name'
+        description: "Model name"
         required: false
-        default: 'all'
+        default: "all"
         type: choice
         options:
           - all
           - torch_cnn_mnist
           - keras_cnn_mnist
       python_version:
-        description: 'Python version'
+        description: "Python version"
         required: false
-        default: 'all'
+        default: "all"
         type: choice
         options:
           - all
-          - '3.10'
-          - '3.11'
-          - '3.12'
+          - "3.10"
+          - "3.11"
+          - "3.12"
       jobs_to_run:
-        description: 'Jobs to run'
+        description: "Jobs to run"
         type: choice
-        default: 'all'
+        default: "all"
         options:
           - all
           - test_with_tls
@@ -139,7 +139,7 @@ jobs:
         uses: actions/[email protected]
         with:
           fetch-depth: 2 # needed for detecting changes
-          submodules: 'true'
+          submodules: "true"
           token: ${{ secrets.GITHUB_TOKEN }}
 
       - name: Pre test run
@@ -152,13 +152,13 @@ jobs:
           python -m pytest -s tests/end_to_end/test_suites/task_runner_tests.py \
           -m task_runner_basic --model_name ${{ env.MODEL_NAME }} \
           --num_rounds ${{ env.NUM_ROUNDS }} --num_collaborators ${{ env.NUM_COLLABORATORS }}
-          echo 'Task runner end to end test run completed'
+          echo "Task runner end to end test run completed"
 
       - name: Post test run
         uses: ./.github/actions/tr_post_test_run
         if: ${{ always() }}
         with:
-          test_type: 'tr_tls'
+          test_type: "tr_tls"
 
   test_with_non_tls:
     name: Test without TLS
@@ -182,7 +182,7 @@ jobs:
         uses: actions/[email protected]
         with:
           fetch-depth: 2 # needed for detecting changes
-          submodules: 'true'
+          submodules: "true"
           token: ${{ secrets.GITHUB_TOKEN }}
 
       - name: Pre test run
@@ -195,13 +195,13 @@ jobs:
           python -m pytest -s tests/end_to_end/test_suites/task_runner_tests.py \
           -m task_runner_basic --model_name ${{ env.MODEL_NAME }} \
           --num_rounds ${{ env.NUM_ROUNDS }} --num_collaborators ${{ env.NUM_COLLABORATORS }} --disable_tls
-          echo 'Task runner end to end test run completed'
+          echo "Task runner end to end test run completed"
 
       - name: Post test run
         uses: ./.github/actions/tr_post_test_run
         if: ${{ always() }}
         with:
-          test_type: 'tr_non_tls'
+          test_type: "tr_non_tls"
 
   test_with_no_client_auth:
     name: Test without client auth
@@ -225,7 +225,7 @@ jobs:
         uses: actions/[email protected]
         with:
           fetch-depth: 2 # needed for detecting changes
-          submodules: 'true'
+          submodules: "true"
           token: ${{ secrets.GITHUB_TOKEN }}
 
       - name: Pre test run
@@ -238,7 +238,7 @@ jobs:
           python -m pytest -s tests/end_to_end/test_suites/task_runner_tests.py \
           -m task_runner_basic --model_name ${{ env.MODEL_NAME }} \
           --num_rounds ${{ env.NUM_ROUNDS }} --num_collaborators ${{ env.NUM_COLLABORATORS }} --disable_client_auth
-          echo 'Task runner end to end test run completed'
+          echo "Task runner end to end test run completed"
 
       - name: Post test run
         uses: ./.github/actions/tr_post_test_run
@@ -268,7 +268,7 @@ jobs:
         uses: actions/[email protected]
         with:
           fetch-depth: 2 # needed for detecting changes
-          submodules: 'true'
+          submodules: "true"
           token: ${{ secrets.GITHUB_TOKEN }}
 
       - name: Pre test run
@@ -282,10 +282,10 @@ jobs:
           -k test_log_memory_usage_basic --model_name ${{ env.MODEL_NAME }} \
           --num_rounds ${{ env.NUM_ROUNDS }} --num_collaborators ${{ env.NUM_COLLABORATORS }} \
           --log_memory_usage
-          echo 'Task runner memory logs test run completed'
+          echo "Task runner memory logs test run completed"
 
       - name: Post test run
         uses: ./.github/actions/tr_post_test_run
         if: ${{ always() }}
         with:
-          test_type: 'tr_tls_memory_logs'
+          test_type: "tr_tls_memory_logs"