Merge pull request #148 from nzw0301/add-missing-experimental

Add missing experimental decorators to the comet integration
optuna · Aug 27, 2024 · b326f52 · b326f52
2 parents 605390c + 2de0e58
commit b326f52
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/optuna_integration/comet/comet.py b/optuna_integration/comet/comet.py
@@ -6,6 +6,7 @@
 import json
 
 import optuna
+from optuna._experimental import experimental_class
 from optuna.study.study import ObjectiveFuncType
 
 from optuna_integration._imports import try_import
@@ -15,6 +16,7 @@
     import comet_ml
 
 
+@experimental_class("4.0.0")
 class CometCallback:
     """
     A callback for logging Optuna study trials to a Comet ML Experiment.