import os
import sys
from daft.daft import connect_start
from pyspark.sql import SparkSession


server = connect_start()
url = f"sc://localhost:{server.port()}"


spark = (
    SparkSession.builder.appName("DaftConnectExample").remote(url).getOrCreate()
)

data = [(1,), (2,), (3,)]
df = spark.createDataFrame(data, ["id"])