Update example and CI file.

Author: ParagEkbote

.github/workflows/comet.yml

@@ -17,38 +17,22 @@ jobs:
         python-version: ['3.8', '3.9', '3.10', '3.11', '3.12']
 
     steps:
-    - name: Checkout Repository
-      uses: actions/checkout@v4
-
-    - name: Setup Python ${{ matrix.python-version }}
+    - uses: actions/checkout@v4
+    - name: setup-python${{ matrix.python-version }}
       uses: actions/setup-python@v5
       with:
         python-version: ${{ matrix.python-version }}
-
-    - name: Install Dependencies
+    - name: Install (Python)
       run: |
         python -m pip install --upgrade pip
         pip install --progress-bar off -U setuptools
         pip install git+https://github.com/optuna/optuna.git
+        python -c 'import optuna'
         pip install git+https://github.com/optuna/optuna-integration.git
+        python -c 'import optuna_integration'
         pip install -r comet/requirements.txt
-
-    - name: Set API Key
+    - name: Run examples
       run: |
-        echo "COMET_API_KEY=${{ secrets.COMET_API_KEY }}" >> $GITHUB_ENV
-
-    - name: Debug API Key
-      run: |
-        echo "Checking COMET_API_KEY..."
-        if [ -z "$COMET_API_KEY" ]; then
-          echo "❌ COMET_API_KEY is NOT set!"
-          exit 1
-        else
-          echo "✅ COMET_API_KEY is set."
-        fi
-
-    - name: Run Examples
-      run: python comet/comet_callback.py
+        python comet/comet_callback.py
       env:
-        OMP_NUM_THREADS: 1
-        COMET_API_KEY: ${{ secrets.COMET_API_KEY }}
+        OMP_NUM_THREADS: 1

comet/comet_callback.py

@@ -15,8 +15,6 @@
     $ python comet_callback.py
 """
 
-import os
-
 import comet_ml
 import optuna
 from optuna_integration.comet import CometCallback
@@ -29,9 +27,7 @@
 from sklearn.model_selection import train_test_split
 
 
-# Ensure the API key is available
-if not os.getenv("COMET_API_KEY"):
-    raise ValueError("COMET_API_KEY is missing! Please set it as an environment variable.")
+comet_ml.login()  # Explicit login
 
 # Start the experiment using comet_ml.start()
 experiment = comet_ml.start()