gistfile1.txt

from enum import Enum

import marvin


class Fruit(str, Enum):
    apple = "apple"
    cherry = "cherry"
    orange = "orange"
    banana = "banana"

@marvin.fn
def rank_fruit(fruit: list[Fruit], order_by: str) -> list[Fruit]:
    """return the indices of the fruit in the enum according to their order"""

if __name__ == "__main__":
    rank_fruit(list(Fruit), "alphabetical")
    # [<Fruit.apple: 'apple'>, <Fruit.banana: 'banana'>, <Fruit.cherry: 'cherry'>, <Fruit.orange: 'orange'>]