From 0682daf536e9c5ec6fb06594fc232b8b9b3c368d Mon Sep 17 00:00:00 2001
From: Pjotr Prins
Date: Wed, 16 Dec 2020 08:40:33 +0000
Subject: Fix checks for `-lmm 9` mode

---
 src/gemma.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'src/gemma.cpp')

diff --git a/src/gemma.cpp b/src/gemma.cpp
index 5a6d4d8..eec9f8f 100644
--- a/src/gemma.cpp
+++ b/src/gemma.cpp
@@ -3383,7 +3383,8 @@ void GEMMA::WriteLog(int argc, char **argv, PARAM &cPar) {
   }
 
   if (cPar.a_mode == 1 || cPar.a_mode == 2 || cPar.a_mode == 3 ||
-      cPar.a_mode == 4 || cPar.a_mode == 5 || cPar.a_mode == 11 ||
+      cPar.a_mode == 4 || cPar.a_mode == 5 || cpar.a_mode == 9 ||
+      cPar.a_mode == 11 ||
       cPar.a_mode == 12 || cPar.a_mode == 13) {
     outfile << "## REMLE log-likelihood in the null model = "
             << cPar.logl_remle_H0 << endl;
@@ -3552,7 +3553,7 @@ void GEMMA::WriteLog(int argc, char **argv, PARAM &cPar) {
     outfile << "##      time on calculating UtX = " << cPar.time_UtX << " min "
             << endl;
   }
-  if ((cPar.a_mode >= 1 && cPar.a_mode <= 4) ||
+  if ((cPar.a_mode >= 1 && cPar.a_mode <= 4) || cPar.a_mode == 9 ||
       (cPar.a_mode >= 51 && cPar.a_mode <= 54)) {
     outfile << "##      time on optimization = " << cPar.time_opt << " min "
             << endl;
-- 
cgit v1.2.3