Skip to content

Commit 3686624

Browse files
authored
Chore: fix python model audit docs (#3632)
1 parent 3c4f7d2 commit 3686624

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

docs/concepts/models/python_models.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -302,6 +302,7 @@ import typing as t
302302
from datetime import datetime
303303

304304
import pandas as pd
305+
from sqlglot.expressions import to_column
305306
from sqlmesh import ExecutionContext, model
306307

307308
@model(
@@ -317,7 +318,7 @@ from sqlmesh import ExecutionContext, model
317318
"name": "Name corresponding to the ID",
318319
},
319320
audits=[
320-
("not_null", {"columns": ["id"]}),
321+
("not_null", {"columns": [to_column("id")]}),
321322
],
322323
)
323324
def execute(

0 commit comments

Comments
 (0)