From 7ecee7821a9c71b23c71772179799e0657458078 Mon Sep 17 00:00:00 2001 From: Bagatur Date: Tue, 12 Sep 2023 15:46:36 -0700 Subject: [PATCH] Replicate fix linting --- libs/langchain/langchain/llms/replicate.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/langchain/langchain/llms/replicate.py b/libs/langchain/langchain/llms/replicate.py index c5b8a8da6a..4ce4621d16 100644 --- a/libs/langchain/langchain/llms/replicate.py +++ b/libs/langchain/langchain/llms/replicate.py @@ -1,5 +1,5 @@ import logging -from typing import Any, Dict, List, Mapping, Optional +from typing import Any, Dict, List, Optional from langchain.callbacks.manager import CallbackManagerForLLMRun from langchain.llms.base import LLM @@ -82,7 +82,7 @@ class Replicate(LLM): return values @property - def _identifying_params(self) -> Mapping[str, Any]: + def _identifying_params(self) -> Dict[str, Any]: """Get the identifying parameters.""" return { "model": self.model, @@ -126,7 +126,7 @@ class Replicate(LLM): self.prompt_key = input_properties[0][0] - inputs = {self.prompt_key: prompt, **self.input} + inputs: Dict = {self.prompt_key: prompt, **self.input} prediction = replicate_python.predictions.create( version=version, input={**inputs, **kwargs}