Replicate fix linting

apify · Sep 12, 2023 · 7ecee78 · 7ecee78
1 parent 21fbbe8
commit 7ecee78
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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 @@ def validate_environment(cls, values: Dict) -> Dict:
         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 @@ def _call(
 
             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}