From 46c09a5b012f4fbc32eac930ff62b2703ca790b1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Kaan=20=C3=87ayl=C4=B1?= Date: Thu, 15 Feb 2024 01:17:59 +0100 Subject: [PATCH] Run black formatter --- app/pipeline/pipeline.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/app/pipeline/pipeline.py b/app/pipeline/pipeline.py index 1379b990..bbcae566 100644 --- a/app/pipeline/pipeline.py +++ b/app/pipeline/pipeline.py @@ -6,19 +6,20 @@ class BasePipeline(metaclass=ABCMeta): - """ Abstract class for all pipelines """ + """Abstract class for all pipelines""" + def __init__(self, name=None): self.name = name def __repr__(self): - return f'{self.__class__.__name__} {self.name if self.name is not None else id(self)}' + return f"{self.__class__.__name__} {self.name if self.name is not None else id(self)}" def __str__(self): - return f'{self.__class__.__name__} {self.name if self.name is not None else id(self)}' + return f"{self.__class__.__name__} {self.name if self.name is not None else id(self)}" @abstractmethod def run(self, *args, **kwargs) -> IrisMessage: - """ Run the pipeline """ + """Run the pipeline""" raise NotImplementedError @@ -29,7 +30,7 @@ def __init__(self, llm, name=None): self.pipeline = {"query": itemgetter("query")} | llm | StrOutputParser() def run(self, *args, query: IrisMessage, **kwargs) -> IrisMessage: - """ A simple pipeline that does not have any memory etc.""" + """A simple pipeline that does not have any memory etc.""" if query is None: raise ValueError("IrisMessage must not be None") message = query.text