Single workflow for TLS and non TLS both

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

.github/workflows/task_runner_e2e.yml

@@ -38,8 +38,8 @@ jobs:
       matrix:
         # There are open issues for some of the models, so excluding them for now:
         # model_name: [ "torch_cnn_mnist", "keras_cnn_mnist", "torch_cnn_histology" ]
-        model_name: ["torch_cnn_mnist", "keras_cnn_mnist"]
-        python_version: ["3.8", "3.9", "3.10"]
+        model_name: ["torch_cnn_mnist"]
+        python_version: ["3.8"]
         tls: [True, False]
       fail-fast: false # do not immediately fail if one of the combinations fail
 
@@ -71,14 +71,14 @@ jobs:
           echo ${{ matrix.tls }}
 
       - name: Run Task Runner E2E tests
-        if: matrix.tls == 'true'
+        if: matrix.tls == true
         id: run_task_runner_tests_tls
         run: |
           python -m pytest -s tests/end_to_end/test_suites/task_runner_tests.py -m ${{ env.MODEL_NAME }} --num_rounds $NUM_ROUNDS --num_collaborators $NUM_COLLABORATORS --model_name ${{ env.MODEL_NAME }}
           echo "Task runner end to end test run completed"
 
       - name: Run Task Runner E2E tests without TLS
-        if: matrix.tls == 'false'
+        if: matrix.tls == false
         id: run_task_runner_tests_non_tls
         run: |
           python -m pytest -s tests/end_to_end/test_suites/task_runner_tests.py -m ${{ env.MODEL_NAME }} --num_rounds $NUM_ROUNDS --num_collaborators $NUM_COLLABORATORS --model_name ${{ env.MODEL_NAME }} --disable_tls