From bbfccae5c56fa236c3cd91ac6bd0077617e7971e Mon Sep 17 00:00:00 2001
From: atalman <atalman@fb.com>
Date: Thu, 28 Dec 2023 12:31:28 -0800
Subject: [PATCH] Add test ops validation for validation workflows

---
 .github/scripts/validate_test_ops.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.github/scripts/validate_test_ops.sh b/.github/scripts/validate_test_ops.sh
index 7c80b601e..d8031c071 100644
--- a/.github/scripts/validate_test_ops.sh
+++ b/.github/scripts/validate_test_ops.sh
@@ -11,6 +11,8 @@ retry git clone --depth 1 https://github.com/pytorch/pytorch.git
 retry git submodule update --init --recursive
 pushd pytorch
 
+pip install expecttest pyyaml jinja2
+
 # Run test_ops validation
 export CUDA_LAUNCH_BLOCKING=1
 python3 test/test_ops.py