updated init

zinggAI · Oct 10, 2024 · 983f4d9 · 983f4d9
1 parent 435f32b
commit 983f4d9
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/spark/core/src/main/java/zingg/spark/core/context/ZinggSparkContext.java b/spark/core/src/main/java/zingg/spark/core/context/ZinggSparkContext.java
@@ -34,10 +34,8 @@ public class ZinggSparkContext extends Context<SparkSession, Dataset<Row>, Row,C
 	protected JavaSparkContext ctx;
     public static final Log LOG = LogFactory.getLog(ZinggSparkContext.class);
 
-
-
-    @Override
-    public void init(SparkSession session)
+
+    public void initSessionAndContext(SparkSession session)
         throws ZinggClientException {
         try{
 //            if (session==null) {
@@ -55,7 +53,6 @@ public void init(SparkSession session)
 				LOG.debug("Context " + ctx.toString());
 				//initHashFns();
 				ctx.setCheckpointDir("/tmp/checkpoint");
-				setUtils();
 			}
         }
         catch(Throwable e) {
@@ -64,6 +61,13 @@ public void init(SparkSession session)
         }
     }
 
+    @Override
+    public void init(SparkSession session)
+        throws ZinggClientException {
+        initSessionAndContext(session);
+		setUtils();
+    }
+
 	@Override
 	public void cleanup() {
 		try {