[MINOR] fix federated transform test
diff --git a/src/test/java/org/apache/sysds/test/AutomatedTestBase.java b/src/test/java/org/apache/sysds/test/AutomatedTestBase.java
index 28a82ff..b0d66f2 100644
--- a/src/test/java/org/apache/sysds/test/AutomatedTestBase.java
+++ b/src/test/java/org/apache/sysds/test/AutomatedTestBase.java
@@ -101,7 +101,7 @@
public static final boolean TEST_GPU = false;
public static final double GPU_TOLERANCE = 1e-9;
- public static final int FED_WORKER_WAIT = 800; // in ms
+ public static final int FED_WORKER_WAIT = 1000; // in ms
// With OpenJDK 8u242 on Windows, the new changes in JDK are not allowing
// to set the native library paths internally thus breaking the code.
diff --git a/src/test/java/org/apache/sysds/test/functions/transform/TransformFederatedEncodeDecodeTest.java b/src/test/java/org/apache/sysds/test/functions/transform/TransformFederatedEncodeDecodeTest.java
index 6a5653e..48a17f5 100644
--- a/src/test/java/org/apache/sysds/test/functions/transform/TransformFederatedEncodeDecodeTest.java
+++ b/src/test/java/org/apache/sysds/test/functions/transform/TransformFederatedEncodeDecodeTest.java
@@ -87,18 +87,18 @@
ExecMode platformOld = rtplatform;
rtplatform = ExecMode.SINGLE_NODE;
- Process t1 = null, t2 = null, t3 = null, t4 = null;
+ Thread t1 = null, t2 = null, t3 = null, t4 = null;
try {
getAndLoadTestConfiguration(TEST_NAME1);
int port1 = getRandomAvailablePort();
- t1 = startLocalFedWorker(port1);
+ t1 = startLocalFedWorkerThread(port1);
int port2 = getRandomAvailablePort();
- t2 = startLocalFedWorker(port2);
+ t2 = startLocalFedWorkerThread(port2);
int port3 = getRandomAvailablePort();
- t3 = startLocalFedWorker(port3);
+ t3 = startLocalFedWorkerThread(port3);
int port4 = getRandomAvailablePort();
- t4 = startLocalFedWorker(port4);
+ t4 = startLocalFedWorkerThread(port4);
// schema
Types.ValueType[] schema = new Types.ValueType[cols / 2];
