We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6426642 commit 46c2b19Copy full SHA for 46c2b19
samples/model-builder/conftest.py
@@ -248,6 +248,17 @@ def mock_run_automl_forecasting_training_job(mock_forecasting_training_job):
248
with patch.object(mock_forecasting_training_job, "run") as mock:
249
yield mock
250
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
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
262
263
@pytest.fixture
264
def mock_get_automl_image_training_job(mock_image_training_job):
0 commit comments