example_csv_output.py

class Node(object):
    def __init__(self, *args, **kwargs):
        self.df: pd.DataFrame = None

    def get_sample(self, *args, n_rows: int = 100, **kwargs):
        return self.df.sample(*args, n=n_rows, **kwargs)

class OutputNode(Node):
    _node_type = "OutputNode"
    pass

class OutputCSVNode(OutputNode):
    _node_subtype = "csv"
    _node_name = "CSV Output"

    def __init__(self, output_path: str, include_index: bool = False):
        super().__init__()
        self.output_path = output_path
        self.include_index = include_index

    def _process(self, input_node: Node):
        self.df = input_node.df
        self.df.to_csv(self.output_path, index=self.include_index)