Add cluster as an arg to GPU workflow (#32)

* add cluster

* gpu type

Author: mvpatel2000

.github/actions/pytest-gpu/action.yaml

@@ -28,6 +28,14 @@ inputs:
   mcloud_api_key:
     required: true
     description: "MosaicML API key - Pass in as repo secret"
+  cluster:
+    required: false
+    type: string
+    default: r1z4
+  gpu_type:
+    required: false
+    type: string
+    default: a100_40gb
   python_version:
     required: false
     type: string
@@ -116,6 +124,8 @@ runs:
         fi
 
         python .github/mcli/mcli_pytest.py \
+          --cluster '${{ inputs.cluster }}' \
+          --gpu_type '${{ inputs.gpu_type }}' \
           --image '${{ inputs.container }}' \
           --git_repo '${{ inputs.git_repo }}' \
           --pip_deps '${{ inputs.pip_deps }}' \