Skip to content

Commit

Permalink
docs(samples): add samples to create/delete featurestore (googleapis#980
Browse files Browse the repository at this point in the history
)

* feat: SDK feature store samples (create/delete fs)

* feat: adding to conftest.py

* docs(samples): fixed testing

* docs(samples): fixed testing

* docs(samples): fixed testing

* docs(samples) added changes

* docs(samples): style issues

* Update samples/model-builder/create_featurestore_sample_test.py

Co-authored-by: Morgan Du <morgandu@google.com>

* Update samples/model-builder/test_constants.py

Co-authored-by: Morgan Du <morgandu@google.com>

* Update samples/model-builder/create_featurestore_sample_test.py

Co-authored-by: Morgan Du <morgandu@google.com>

Co-authored-by: Morgan Du <morgandu@google.com>
  • Loading branch information
nayaknishant and morgandu committed Feb 24, 2022
1 parent 41d8fdd commit b19fbf0
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions samples/model-builder/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -387,9 +387,12 @@ def mock_get_featurestore(mock_featurestore):


@pytest.fixture
def mock_create_featurestore():
with patch.object(aiplatform.Featurestore, "create") as mock:
yield mock
def mock_create_featurestore(mock_featurestore):
with patch.object(
aiplatform.featurestore.Featurestore, "create"
) as mock_create_featurestore:
mock_create_featurestore.return_value = mock_featurestore
yield mock_create_featurestore


@pytest.fixture
Expand Down

0 comments on commit b19fbf0

Please sign in to comment.