From c5cd0d164b6f84ea4b4c7b26da7ec9aa4204acc5 Mon Sep 17 00:00:00 2001
From: Jirka Borovec <jirka.borovec@seznam.cz>
Date: Tue, 8 Dec 2020 10:10:28 +0100
Subject: [PATCH] skip hanging

---
 tests/models/test_scripts.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/models/test_scripts.py b/tests/models/test_scripts.py
index 56f396cbea..55f9b14d19 100644
--- a/tests/models/test_scripts.py
+++ b/tests/models/test_scripts.py
@@ -87,6 +87,8 @@ def test_cli_run_log_regression(cli_args):
         cli_main()
 
 
+# TODO: this test is hanging (runs for more then 10min) so we need to use GPU or optimize it...
+@pytest.mark.skipif(not torch.cuda.is_available(), reason="test requires GPU machine")
 @pytest.mark.parametrize('cli_args', [f'--data_dir {DATASETS_PATH} --max_epochs 1 --max_steps 2'])
 def test_cli_run_vision_image_gpt(cli_args):
     """Test running CLI for an example with default params."""