Skip to content

Commit d49b795

Browse files
committed
Fix integration tests
1 parent 5b93624 commit d49b795

2 files changed

Lines changed: 3 additions & 1 deletion

File tree

tests/functional/adapter/python_model/fixtures.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -354,6 +354,7 @@ def model(dbt, spark):
354354
schema="id int, data string",
355355
data=[(1, "a"), (2, "b")]
356356
)
357+
return df
357358
"""
358359

359360
# Notebook-scoped packages via notebook job run (all_purpose_cluster, create_notebook=True)
@@ -372,4 +373,5 @@ def model(dbt, spark):
372373
schema="id int, data string",
373374
data=[(1, "a"), (2, "b")]
374375
)
376+
return df
375377
"""

tests/functional/adapter/python_model/test_python_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -470,7 +470,7 @@ class TestNotebookScopedPackagesCommandAPI:
470470

471471
@pytest.fixture(scope="class")
472472
def models(self):
473-
model = override_fixtures.notebook_scoped_packages_notebook_run_model
473+
model = override_fixtures.notebook_scoped_packages_cmd_api_model
474474
return {"notebook_scoped_packages_cmd.py": model}
475475

476476
def test_notebook_scoped_packages_command_api(self, project):

0 commit comments

Comments
 (0)