1from __future__ import annotations
2
3import os
4import daft
5from daft import col
6
7from daft.aihc.common.udf import aihc_udf
8from daft.aihc.functions.text.md5_calculator import Md5Calculator
9
10if __name__ == "__main__":
11    if os.getenv("DAFT_RUNNER", "native") == "ray":
12        import ray
13        ray.init(dashboard_host="0.0.0.0", ignore_reinit_error=True)
14        daft.set_runner_ray()
15    daft.set_execution_config(actor_udf_ready_timeout=6000, min_cpu_per_task=0)
16
17    samples = {
18        "text": [
19            'Hello, World!',
20            "Hello, World !",
21            'Hello,World!',
22        ]
23    }
24
25    ds = daft.from_pydict(samples)
26    ds = ds.with_column(
27        "md5_hash",
28        aihc_udf(Md5Calculator)(col("text")),
29    )
30
31    ds.show()