Skip to content

Commit 46c2b19

Browse files
committed
Fixed test and mocks
1 parent 6426642 commit 46c2b19

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

samples/model-builder/conftest.py

+11
Original file line numberDiff line numberDiff line change
@@ -248,6 +248,17 @@ def mock_run_automl_forecasting_training_job(mock_forecasting_training_job):
248248
with patch.object(mock_forecasting_training_job, "run") as mock:
249249
yield mock
250250

251+
@pytest.fixture
252+
def mock_get_automl_forecasting_seq2seq_training_job(mock_forecasting_training_job):
253+
with patch.object(aiplatform, "SequenceToSequencePlusForecastingTrainingJob") as mock:
254+
mock.return_value = mock_forecasting_training_job
255+
yield mock
256+
257+
258+
@pytest.fixture
259+
def mock_run_automl_forecasting_seq2seq_training_job(mock_forecasting_training_job):
260+
with patch.object(mock_forecasting_training_job, "run") as mock:
261+
yield mock
251262

252263
@pytest.fixture
253264
def mock_get_automl_image_training_job(mock_image_training_job):

0 commit comments

Comments
 (0)