Run dockerhub and ecr together

Signed-off-by: Divya Madala <divyaasm@amazon.com>
Divyaasm · Jun 11, 2024 · d87af0e · d87af0e
1 parent 59be78e
commit d87af0e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/run_validation.py b/src/run_validation.py
@@ -21,7 +21,12 @@ def main() -> int:
     logging.getLogger("urllib3").setLevel(logging.WARNING)
 
     with TemporaryDirectory() as work_dir:
-        test_result = ValidationTestRunner.dispatch(args, args.distribution, work_dir).run()
+        if args.distribution == "docker":
+            test_result = ValidationTestRunner.dispatch(args, args.distribution, work_dir)
+            for result in test_result:
+                test_result = test_result.run()
+        else:
+            test_result = ValidationTestRunner.dispatch(args, args.distribution, work_dir).run()
         logging.info(f'final test_result = {test_result}')
         return 0 if test_result else 1  # type: ignore