feat: allow the learner to inherit in measure assertion

m-muecke · m-muecke · commit 118ae657277b · 2025-12-03T18:47:54.000+01:00
diff --git a/R/assertions.R b/R/assertions.R
@@ -259,7 +259,7 @@ assert_measure = function(measure, task = NULL, learner = NULL, prediction = NUL
 
   if (!is.null(learner)) {
 
-    if (!is_scalar_na(measure$task_type) && measure$task_type != learner$task_type) {
+    if (!is_scalar_na(measure$task_type) && !test_matching_task_type(measure$task_type, learner, "learner")) {
       error_input("Measure '%s' is not compatible with type '%s' of learner '%s'",
         measure$id, learner$task_type, learner$id)
     }

Original file line number	Diff line number	Diff line change
`@@ -259,7 +259,7 @@ assert_measure = function(measure, task = NULL, learner = NULL, prediction = NUL`
`259`	`259`
`260`	`260`	`if (!is.null(learner)) {`
`261`	`261`
`262`		`- if (!is_scalar_na(measure$task_type) && measure$task_type != learner$task_type) {`
	`262`	`+ if (!is_scalar_na(measure$task_type) && !test_matching_task_type(measure$task_type, learner, "learner")) {`
`263`	`263`	`error_input("Measure '%s' is not compatible with type '%s' of learner '%s'",`
`264`	`264`	`measure$id, learner$task_type, learner$id)`
`265`	`265`	`}`