suggestion for binary check

Author: PhilippBach

R/helper.R

@@ -229,11 +229,11 @@ initiate_task = function(id, data, target, select_cols, learner_class) {
     task = TaskRegr$new(id = id, backend = data, target = target)
   } else if (learner_class == "LearnerClassif") {
     data[[target]] = factor(data[[target]])
+    assert_set_equal(levels(data[[target]]),
+                     c("0", "1"))
     task = TaskClassif$new(
       id = id, backend = data, target = target,
       positive = "1")
-    assert_set_equal(c(task$positive, task$negative),
-                     c("0", "1"))
   }
   return(task)
 }