Update mglmxls.jl

mwsohn · Oct 6, 2024 · 34dd400 · 34dd400
1 parent a7ef8c3
commit 34dd400
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/mglmxls.jl b/src/mglmxls.jl
@@ -208,17 +208,20 @@ function mglmxls(glmout,
 
         vn = covars[i]
 
+        # count the number of levels in a categorical variable
+        if haskey(vvalues, vn)
+            nlev[i] = length(vvalues[vn])
+        end
+
         # use labels if exist
         if labels != nothing && haskey(labels, Symbol(vn))
             varname[i] = labels[Symbol(vn)]
+        elseif nlev[i] == 2 && occursin(": 1",vn)
+            varname[i] = chop(vn, tail=3)
         else
             varname[i] = vn
         end
 
-        # count the number of levels in a categorical variable
-        if haskey(vvalues, vn)
-            nlev[i] = length(vvalues[vn])
-        end
     end
 
     # write table