Fixed test and mocks

Author: ivanmkc

samples/model-builder/conftest.py

@@ -248,6 +248,17 @@ def mock_run_automl_forecasting_training_job(mock_forecasting_training_job):
     with patch.object(mock_forecasting_training_job, "run") as mock:
         yield mock
 
+@pytest.fixture
+def mock_get_automl_forecasting_seq2seq_training_job(mock_forecasting_training_job):
+    with patch.object(aiplatform, "SequenceToSequencePlusForecastingTrainingJob") as mock:
+        mock.return_value = mock_forecasting_training_job
+        yield mock
+
+
+@pytest.fixture
+def mock_run_automl_forecasting_seq2seq_training_job(mock_forecasting_training_job):
+    with patch.object(mock_forecasting_training_job, "run") as mock:
+        yield mock
 
 @pytest.fixture
 def mock_get_automl_image_training_job(mock_image_training_job):

samples/model-builder/create_training_pipeline_forecasting_seq2seq_sample_test.py

@@ -20,8 +20,8 @@
 def test_create_training_pipeline_forecasting_seq2seq_sample(
     mock_sdk_init,
     mock_time_series_dataset,
-    mock_get_automl_forecasting_seq2seq_training_job,
-    mock_run_automl_forecasting_seq2seq_training_job,
+    mock_get_automl_forecasting_training_job,
+    mock_run_automl_forecasting_training_job,
     mock_get_time_series_dataset,
 ):
 
@@ -52,11 +52,11 @@ def test_create_training_pipeline_forecasting_seq2seq_sample(
     mock_sdk_init.assert_called_once_with(
         project=constants.PROJECT, location=constants.LOCATION
     )
-    mock_get_automl_forecasting_seq2seq_training_job.assert_called_once_with(
+    mock_get_automl_forecasting_training_job.assert_called_once_with(
         display_name=constants.DISPLAY_NAME,
         optimization_objective="minimize-rmse",
     )
-    mock_run_automl_forecasting_seq2seq_training_job.assert_called_once_with(
+    mock_run_automl_forecasting_training_job.assert_called_once_with(
         dataset=mock_time_series_dataset,
         target_column=constants.TABULAR_TARGET_COLUMN,
         time_column=constants.FORECASTNG_TIME_COLUMN,