From 601f4c8f6cb9ba107773a414b5b86ac7157a5740 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..c05b5dfca 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()).type_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()).type_dict ``` ````