Update example.

Author: ParagEkbote

.github/workflows/comet.yml

@@ -32,6 +32,8 @@ jobs:
         python -c 'import optuna_integration'
 
         pip install -r comet/requirements.txt
+    - name: Set API Key
+      run: echo "COMET_API_KEY=${{ secrets.COMET_API_KEY }}" >> $GITHUB_ENV
     - name: Run examples
       run: |
         python comet/comet_callback.py

comet/comet_callback.py

@@ -17,6 +17,7 @@
 
 import comet_ml
 import optuna
+import os
 from optuna_integration.comet import CometCallback
 
 from sklearn.datasets import load_breast_cancer
@@ -27,8 +28,8 @@
 from sklearn.model_selection import train_test_split
 
 
-# Initialize Comet ML
-comet_ml.login(project_name="comet-optuna-sklearn-example")
+experiment = comet_ml.Experiment(api_key=os.getenv("COMET_API_KEY"))
+comet_ml.login(project_name="comet-optuna-example")
 experiment = comet_ml.start()
 
 # Load dataset