marondeau commited on
Commit
9f298e3
·
unverified ·
1 Parent(s): cb9fd37

Remove clever booleanization of Response

Browse files
Files changed (1) hide show
  1. buster/formatter/base.py +3 -6
buster/formatter/base.py CHANGED
@@ -17,9 +17,6 @@ class Response:
17
  error: bool = False
18
  error_msg: str | None = None
19
 
20
- def __bool__(self) -> bool:
21
- return not self.error
22
-
23
 
24
  @dataclass
25
  class Formatter:
@@ -58,6 +55,6 @@ class Formatter:
58
 
59
  def __call__(self, response: Response, sources: Iterable[Source]) -> str:
60
  """Format an answer and its sources, or an error message."""
61
- if response:
62
- return self.answer(response, sources)
63
- return self.error(response)
 
17
  error: bool = False
18
  error_msg: str | None = None
19
 
 
 
 
20
 
21
  @dataclass
22
  class Formatter:
 
55
 
56
  def __call__(self, response: Response, sources: Iterable[Source]) -> str:
57
  """Format an answer and its sources, or an error message."""
58
+ if response.error:
59
+ return self.error(response)
60
+ return self.answer(response, sources)