From 9b3668018c52e16a863eea557722b2e48a8f33c8 Mon Sep 17 00:00:00 2001
From: mibe <mikhail.beck@exasol.com>
Date: Thu, 1 Feb 2024 15:49:13 +0000
Subject: [PATCH] Trying to get the translation test working

---
 test/notebooks/nbtest_itde.py         |  3 +++
 test/notebooks/nbtest_sagemaker.py    |  2 ++
 test/notebooks/nbtest_sklearn.py      |  3 +++
 test/notebooks/nbtest_transformers.py | 18 ++++++++++++------
 4 files changed, 20 insertions(+), 6 deletions(-)

diff --git a/test/notebooks/nbtest_itde.py b/test/notebooks/nbtest_itde.py
index 6a9b8b02..0a36b19e 100644
--- a/test/notebooks/nbtest_itde.py
+++ b/test/notebooks/nbtest_itde.py
@@ -1,3 +1,4 @@
+import pytest
 from exasol.secret_store import Secrets
 from exasol.itde_manager import (
     bring_itde_up,
@@ -5,6 +6,8 @@
 )
 from exasol.connections import open_pyexasol_connection
 
+
+@pytest.mark.skip('just want to see if the translation works')
 def test_itde(tmp_path):
     store_path = tmp_path / 'tmp_config.sqlite'
     store_password = "password"
diff --git a/test/notebooks/nbtest_sagemaker.py b/test/notebooks/nbtest_sagemaker.py
index a410006f..c1a27c1c 100644
--- a/test/notebooks/nbtest_sagemaker.py
+++ b/test/notebooks/nbtest_sagemaker.py
@@ -5,6 +5,7 @@
 from inspect import cleandoc
 import textwrap
 
+import pytest
 from exasol.secret_store import Secrets
 from exasol.ai_lab_config import AILabConfig as CKey
 
@@ -172,6 +173,7 @@ def continuous_job_polling():
     )
 
 
+@pytest.mark.skip('just want to see if the translation works')
 def test_sagemaker(access_to_temp_secret_store, uploading_hack):
 
     store_path, store_password = access_to_temp_secret_store
diff --git a/test/notebooks/nbtest_sklearn.py b/test/notebooks/nbtest_sklearn.py
index 60c93faa..e52960a2 100644
--- a/test/notebooks/nbtest_sklearn.py
+++ b/test/notebooks/nbtest_sklearn.py
@@ -1,8 +1,10 @@
 import os
+import pytest
 
 from notebook_test_utils import (access_to_temp_secret_store, notebook_runner)
 
 
+@pytest.mark.skip('just want to see if the translation works')
 def test_regression(notebook_runner) -> None:
 
     current_dir = os.getcwd()
@@ -18,6 +20,7 @@ def test_regression(notebook_runner) -> None:
         os.chdir(current_dir)
 
 
+@pytest.mark.skip('just want to see if the translation works')
 def test_classification(notebook_runner) -> None:
 
     current_dir = os.getcwd()
diff --git a/test/notebooks/nbtest_transformers.py b/test/notebooks/nbtest_transformers.py
index a05fa43e..c1e853a2 100644
--- a/test/notebooks/nbtest_transformers.py
+++ b/test/notebooks/nbtest_transformers.py
@@ -8,13 +8,19 @@
 @pytest.mark.parametrize(
     "notebook_file",
     [
-        'sequence_classification.ipynb',
-        'question_answering.ipynb',
-        'masked_modelling.ipynb',
-        'token_classification.ipynb',
-        'text_generation.ipynb',
+        pytest.param('sequence_classification.ipynb',
+                     marks=pytest.mark.xfail(reason='just want to see if the translation works')),
+        pytest.param('question_answering.ipynb',
+                     marks=pytest.mark.xfail(reason='just want to see if the translation works')),
+        pytest.param('masked_modelling.ipynb',
+                     marks=pytest.mark.xfail(reason='just want to see if the translation works')),
+        pytest.param('token_classification.ipynb',
+                     marks=pytest.mark.xfail(reason='just want to see if the translation works')),
+        pytest.param('text_generation.ipynb',
+                     marks=pytest.mark.xfail(reason='just want to see if the translation works')),
         'translation.ipynb',
-        'zero_shot_classification.ipynb'
+        pytest.param('zero_shot_classification.ipynb',
+                     marks=pytest.mark.xfail(reason='just want to see if the translation works')),
     ]
 )
 def test_transformers(notebook_runner, uploading_hack, notebook_file) -> None: