fix test error.

h2oai · Oct 27, 2024 · cbe0444 · cbe0444
1 parent e877f21
commit cbe0444
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/h2o-algos/src/main/java/hex/hglm/ComputationStateHGLM.java b/h2o-algos/src/main/java/hex/hglm/ComputationStateHGLM.java
@@ -79,7 +79,7 @@ void initComputationStateHGLM(HGLMTask.ComputationEngineTask engineTask) {
         throw new IllegalArgumentException("initial_t_matrix must be symmetric but is not!");
       // make sure matrix is semi positive definite
       Matrix tMat = new Matrix(_T);
-      if (!tMat.chol().isSPD())
+      if ((_parms._max_iterations > 0) && !tMat.chol().isSPD()) // only check this when we actually build the model
         throw new IllegalArgumentException("initial_t_matrix must be positive semi definite but is not!");
     } else {
       if (_parms._tau_u_var_init > 0.0) {