Skip to content

Commit

Permalink
Fixed test and mocks
Browse files Browse the repository at this point in the history
  • Loading branch information
ivanmkc committed Aug 16, 2022
1 parent 6426642 commit 46c2b19
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions samples/model-builder/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 46c2b19

Please sign in to comment.