From c32b46c130fa8395ddfcc6a4216d50761be9ecc6 Mon Sep 17 00:00:00 2001
From: Joaquin Anton Guirao <janton@nvidia.com>
Date: Thu, 9 Jan 2025 13:21:06 +0100
Subject: [PATCH] Move optical flow tests from specific TU test job to Ampere
 tests

Signed-off-by: Joaquin Anton Guirao <janton@nvidia.com>
---
 qa/TL0_python_self_test_TU/test.sh | 18 ------------------
 qa/TL0_self_test_Ampere/test.sh    |  4 ++++
 2 files changed, 4 insertions(+), 18 deletions(-)
 delete mode 100755 qa/TL0_python_self_test_TU/test.sh

diff --git a/qa/TL0_python_self_test_TU/test.sh b/qa/TL0_python_self_test_TU/test.sh
deleted file mode 100755
index 4da120a6027..00000000000
--- a/qa/TL0_python_self_test_TU/test.sh
+++ /dev/null
@@ -1,18 +0,0 @@
-#!/bin/bash -e
-# used pip packages
-pip_packages='${python_test_runner_package} numpy opencv-python pillow nvidia-ml-py==11.450.51 torch numba'
-target_dir=./dali/test/python
-
-# populate epilog and prolog with variants to enable/disable conda
-# every test will be executed for bellow configs
-prolog=(: enable_conda)
-epilog=(: disable_conda)
-
-test_body() {
-    ${python_invoke_test} test_optical_flow.py
-    ${python_invoke_test} test_dali_variable_batch_size.py:test_optical_flow
-}
-
-pushd ../..
-source ./qa/test_template.sh
-popd
diff --git a/qa/TL0_self_test_Ampere/test.sh b/qa/TL0_self_test_Ampere/test.sh
index e48a4619b68..0167fa407c3 100644
--- a/qa/TL0_self_test_Ampere/test.sh
+++ b/qa/TL0_self_test_Ampere/test.sh
@@ -31,6 +31,10 @@ test_body() {
 
   # test decoders on A100 as well
   ${python_new_invoke_test} -s decoder test_image
+
+  # test Optical Flow
+  ${python_invoke_test} test_optical_flow.py
+  ${python_invoke_test} test_dali_variable_batch_size.py:test_optical_flow
 }
 
 pushd ../..