Merge pull request #63 from ehrlinger/copilot_refactor

ehrlinger · web-flow · commit ffb7f9a6d480 · 2026-03-16T15:26:21.000-04:00
Fix model label assignment in gg_partial function for categorical data
diff --git a/R/gg_partial.R b/R/gg_partial.R
@@ -104,7 +104,10 @@ gg_partial <- function(part_dta,
   }
   ## Optionally attach a model label (useful when overlaying multiple forests)
   if (!is.null(model)) {
-    continuous$model <- categorical$model <- model
+    continuous$model <- model
+    if(!is.null(categorical) && nrow(categorical) > 0) {
+      categorical$model <- model
+    }
   }
 
   return(list(continuous = continuous, categorical = categorical))

Original file line number	Diff line number	Diff line change
`@@ -104,7 +104,10 @@ gg_partial <- function(part_dta,`
`104`	`104`	`}`
`105`	`105`	`## Optionally attach a model label (useful when overlaying multiple forests)`
`106`	`106`	`if (!is.null(model)) {`
`107`		`- continuous$model <- categorical$model <- model`
	`107`	`+ continuous$model <- model`
	`108`	`+ if(!is.null(categorical) && nrow(categorical) > 0) {`
	`109`	`+ categorical$model <- model`
	`110`	`+ }`
`108`	`111`	`}`
`109`	`112`
`110`	`113`	`return(list(continuous = continuous, categorical = categorical))`