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.audio.audio_duration import AudioDuration
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        "audio_path": [
19            "file:///local/sample_1.wav",
20            "file:///mnt/pfs/sample_2.wav",
21            "file:///mnt/bos/sample_3.wav",
22        ]
23    }
24    
25    df = daft.from_pydict(samples)
26    df = df.with_column("duration_result", aihc_udf(AudioDuration)(col("audio_path")))
27    df.show()