From ffbb1046485159fc4493d6dc6ab0f25a73c04ee8 Mon Sep 17 00:00:00 2001 From: ayush chaurasia Date: Tue, 16 Apr 2024 09:51:08 +0530 Subject: [PATCH] remove protected namespaces --- python/python/lancedb/embeddings/fine_tuner/llm.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python/python/lancedb/embeddings/fine_tuner/llm.py b/python/python/lancedb/embeddings/fine_tuner/llm.py index 1bdf2d97..510c1fea 100644 --- a/python/python/lancedb/embeddings/fine_tuner/llm.py +++ b/python/python/lancedb/embeddings/fine_tuner/llm.py @@ -3,7 +3,7 @@ import re from functools import cached_property from typing import Optional -from pydantic import BaseModel +from pydantic import BaseModel, config from ...util import attempt_import_or_raise from ..utils import api_key_not_found_help @@ -15,10 +15,12 @@ class BaseLLM(BaseModel): Base class for Language Model based Embedding Functions. This class is loosely desined rn, and will be updated as the usage gets clearer. """ + class Config: + protected_namespaces = () # Disable protected namespace check model_name: str model_kwargs: dict = {} - + @cached_property def _client(): """