From ca2eab0540e3ca7996b2dc00011900f5a5e27369 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Edgar=20Ram=C3=ADrez-Mondrag=C3=B3n?= Date: Sat, 24 Aug 2024 13:14:57 -0600 Subject: [PATCH] Fix example in docs --- docs/guides/sql-tap.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/guides/sql-tap.md b/docs/guides/sql-tap.md index 5cbb8b310..4609e2721 100644 --- a/docs/guides/sql-tap.md +++ b/docs/guides/sql-tap.md @@ -31,7 +31,7 @@ class CustomSQLToJSONSchema(SQLToJSONSchema): @SQLToJSONSchema.to_jsonschema.register(VectorType) def vector_to_json_schema(self, column_type): """Custom vector to JSON schema.""" - return th.ArrayType(items=th.NumberType()) + return th.ArrayType(th.NumberType()).to_dict() ``` ````{tip} @@ -40,7 +40,7 @@ You can also use a type annotation to specify the type of the column when regist ```python @SQLToJSONSchema.to_jsonschema.register def vector_to_json_schema(self, column_type: VectorType): - return th.ArrayType(items=th.NumberType()) + return th.ArrayType(th.NumberType()).to_dict() ``` ````