using ZinggWithSpark in python causes error zinggAI#640

zinggAI · Aug 9, 2023 · 4fd68f9 · 4fd68f9
1 parent d4789fd
commit 4fd68f9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/spark/client/src/main/java/zingg/spark/client/SparkClient.java b/spark/client/src/main/java/zingg/spark/client/SparkClient.java
@@ -6,6 +6,7 @@
 import org.apache.spark.sql.Dataset;
 import org.apache.spark.sql.Row;
 import org.apache.spark.sql.types.DataType;
+import org.apache.spark.sql.SparkSession;
 
 import zingg.common.client.Arguments;
 import zingg.common.client.Client;
@@ -33,6 +34,10 @@ public SparkClient(Arguments args, ClientOptions options, ZSparkSession s) throw
 		super(args, options, s);
 	}
 
+	public SparkClient(Arguments args, ClientOptions options, SparkSession s) throws ZinggClientException {
+		this(args, options, new ZSparkSession(s,null));
+	}
+
 	public SparkClient() {
 		/*SparkSession session = SparkSession
                 .builder()