Skip to content

Commit 34307c6

Browse files
GarrettWuGenesis929
authored andcommitted
chore: fix model.register test to use anonymous dataset (#510)
1 parent cc326ec commit 34307c6

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tests/system/small/ml/conftest.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,12 +47,11 @@ def penguins_bqml_linear_model(session, penguins_linear_model_name) -> core.Bqml
4747

4848
@pytest.fixture(scope="function")
4949
def ephemera_penguins_bqml_linear_model(
50-
penguins_bqml_linear_model,
50+
session: bigframes.Session,
51+
penguins_bqml_linear_model: core.BqmlModel,
5152
) -> core.BqmlModel:
5253
model = penguins_bqml_linear_model
53-
return model.copy(
54-
f"{model._model.project}.{model._model.dataset_id}.{uuid.uuid4().hex}"
55-
)
54+
return model.copy(f"{session._anonymous_dataset}.{uuid.uuid4().hex}")
5655

5756

5857
@pytest.fixture(scope="session")

0 commit comments

Comments
 (0)