From dcaea97f8390f1b09fd5080bc892b813fadb7b4c Mon Sep 17 00:00:00 2001
From: George Sittas <giwrgos.sittas@gmail.com>
Date: Tue, 14 Jan 2025 19:05:43 +0200
Subject: [PATCH] Chore: fix python model audit docs

---
 docs/concepts/models/python_models.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/docs/concepts/models/python_models.md b/docs/concepts/models/python_models.md
index a66a5649f4..a5380ef6f6 100644
--- a/docs/concepts/models/python_models.md
+++ b/docs/concepts/models/python_models.md
@@ -302,6 +302,7 @@ import typing as t
 from datetime import datetime
 
 import pandas as pd
+from sqlglot.expressions import to_column
 from sqlmesh import ExecutionContext, model
 
 @model(
@@ -317,7 +318,7 @@ from sqlmesh import ExecutionContext, model
         "name": "Name corresponding to the ID",
     },
     audits=[
-        ("not_null", {"columns": ["id"]}),
+        ("not_null", {"columns": [to_column("id")]}),
     ],
 )
 def execute(