From 73da8f863cff5ef6af358ec173572a1a8495baef Mon Sep 17 00:00:00 2001 From: Harutaka Kawamura Date: Wed, 3 Jan 2024 15:45:18 +0900 Subject: [PATCH] Remove unused `Params` (#14385) Removes unused `Params` in `libs/langchain/langchain/llms/mlflow.py`. Co-authored-by: Harrison Chase --- libs/community/langchain_community/llms/mlflow.py | 13 +------------ 1 file changed, 1 insertion(+), 12 deletions(-) diff --git a/libs/community/langchain_community/llms/mlflow.py b/libs/community/langchain_community/llms/mlflow.py index b8f733d720d..0110e4b40e0 100644 --- a/libs/community/langchain_community/llms/mlflow.py +++ b/libs/community/langchain_community/llms/mlflow.py @@ -5,18 +5,7 @@ from urllib.parse import urlparse from langchain_core.callbacks import CallbackManagerForLLMRun from langchain_core.language_models import LLM -from langchain_core.pydantic_v1 import BaseModel, Extra, Field, PrivateAttr - - -# Ignoring type because below is valid pydantic code -# Unexpected keyword argument "extra" for "__init_subclass__" of "object" -class Params(BaseModel, extra=Extra.allow): # type: ignore[call-arg] - """Parameters for MLflow""" - - temperature: float = 0.0 - n: int = 1 - stop: Optional[List[str]] = None - max_tokens: Optional[int] = None +from langchain_core.pydantic_v1 import Field, PrivateAttr class Mlflow(LLM):