Fix groups logic for use_child_oof (#4111)

autogluon · Apr 17, 2024 · 04a390c · 04a390c
1 parent 8e06686
commit 04a390c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/core/src/autogluon/core/models/ensemble/bagged_ensemble_model.py b/core/src/autogluon/core/models/ensemble/bagged_ensemble_model.py
@@ -197,6 +197,9 @@ def _fit(
         **kwargs,
     ):
         use_child_oof = self.params.get("use_child_oof", False)
+        if use_child_oof and groups is not None:
+            logger.log(20, f"\tForcing `use_child_oof=False` because `groups` is specified")
+            use_child_oof = False
         if use_child_oof:
             if self.is_fit():
                 # TODO: We may want to throw an exception instead and avoid calling fit more than once