Stefano Fiorucci
819ed7244b
* draft * progress * more fixes * improvs * strict * rm pylint * rm pylint comments * fix and simplify * standardize * more * alpha order * final touch * remove template * rm ellipsis comments
23 lines
576 B
Python
23 lines
576 B
Python
# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from haystack.core.component import component
|
|
|
|
|
|
@component
|
|
class Parity:
|
|
"""
|
|
Redirects the value, unchanged, along the 'even' connection if even, or along the 'odd' one if odd.
|
|
"""
|
|
|
|
@component.output_types(even=int, odd=int)
|
|
def run(self, value: int):
|
|
"""
|
|
:param value: The value to check for parity
|
|
"""
|
|
remainder = value % 2
|
|
if remainder:
|
|
return {"odd": value}
|
|
return {"even": value}
|