Skip to content
This repository was archived by the owner on May 7, 2026. It is now read-only.
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix mypy
  • Loading branch information
GarrettWu committed Mar 8, 2024
commit 2b1464fab87012b4026a5a2d368914a95112d02c
1 change: 1 addition & 0 deletions tests/system/small/ml/test_core.py
Original file line number Diff line number Diff line change
Expand Up @@ -394,6 +394,7 @@ def test_model_register(ephemera_penguins_bqml_linear_model: core.BqmlModel):
model = ephemera_penguins_bqml_linear_model
model.register()

assert model.model.model_id is not None
model_name = "bigframes_" + model.model.model_id
# Only registered model contains the field, and the field includes project/dataset. Here only check model_id.
assert model_name in model.model.training_runs[-1]["vertexAiModelId"]
Expand Down