chore: fix tpch data generator (apache#1186)

milenkovicm · Feb 19, 2025 · 488893b · 488893b
1 parent 3578245
commit 488893b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/benchmarks/tpch-gen.sh b/benchmarks/tpch-gen.sh
@@ -29,13 +29,13 @@ FILE=./data/supplier.tbl
 if test -f "$FILE"; then
     echo "$FILE exists."
 else
-  docker run -v `pwd`/data:/data -it --rm ghcr.io/databloom-ai/tpch-docker:main -vf -s 1
+  docker run -v `pwd`/data:/data -it --rm ghcr.io/scalytics/tpch-docker:main -vf -s 1
 fi
 
 # Copy expected answers into the ./data/answers directory if it does not already exist
 FILE=./data/answers/q1.out
 if test -f "$FILE"; then
     echo "$FILE exists."
 else
-  docker run -v `pwd`/data:/data -it --entrypoint /bin/bash --rm ghcr.io/databloom-ai/tpch-docker:main -c "cp /opt/tpch/2.18.0_rc2/dbgen/answers/* /data/answers/"
+  docker run -v `pwd`/data:/data -it --entrypoint /bin/bash --rm ghcr.io/scalytics/tpch-docker:main -c "cp /opt/tpch/2.18.0_rc2/dbgen/answers/* /data/answers/"
 fi
diff --git a/benchmarks/tpch.py b/benchmarks/tpch.py
@@ -30,8 +30,10 @@
 path = args.path
 table_ext = args.ext
 
-import ballista
-ctx = ballista.BallistaContext("localhost", 50050)
+from ballista import BallistaBuilder
+from datafusion.context import SessionContext
+
+ctx: SessionContext = BallistaBuilder().remote("df://127.0.0.1:50050")
 
 tables = ["part", "supplier", "partsupp", "customer", "orders", "lineitem", "nation", "region"]