small fix

Author: gs-olive

.github/workflows/docker_builder.yml

@@ -2,10 +2,12 @@ name: 'Torch-TensorRT Docker Build'
 
 # Apply workflow only to main branch
 on:
-  push:
-    branches:
-      - main
-      - nightly
+  pull_request:
+    types: [opened, synchronize, ready_for_review, review_requested, reopened]
+  # push:
+  #   branches:
+  #     - main
+  #     - nightly
 
 # If pushes to main are made in rapid succession,
 # cancel existing docker builds and use newer commits
@@ -21,7 +23,8 @@ jobs:
     # Container name is of the form torch_tensorrt:<branch_name>
     env:
       DOCKER_REGISTRY: ghcr.io/pytorch/tensorrt
-      CONTAINER_NAME: torch_tensorrt:${{ github.ref_name }}
+      CONTAINER_NAME: torch_tensorrt:main
+      #${{ github.ref_name }}
 
     steps:
     - name: Checkout repository
@@ -39,6 +42,7 @@ jobs:
       env:
         DOCKER_TAG: ${{ env.DOCKER_REGISTRY }}/${{ env.CONTAINER_NAME }}
       run: |
+        python3 -m pip install pyyaml
         TRT_VERSION=$(python3 -c "import versions; versions.tensorrt_version()")
         echo "TRT VERSION = ${TRT_VERSION}"
         CUDNN_VERSION=$(python3 -c "import versions; versions.cudnn_version()")