fix emmissions

tasks/text.py

@@ -10,7 +10,7 @@ import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
 from .utils.evaluation import TextEvaluationRequest
-from .utils.emissions import tracker, clean_emissions_data, get_space_info
+from .utils.emissions import tracker, clean_emissions_data, get_space_info, start_tracking, stop_tracking
 
 # Disable torch compile
 os.environ["TORCH_COMPILE_DISABLE"] = "1"
@@ -112,8 +112,9 @@ async def evaluate_text(request: TextEvaluationRequest):
     test_dataset = dataset["test"]
     
     # Start tracking emissions
-    tracker.start()
-    tracker.start_task("inference")
+    start_tracking()
+
+    # tracker.start_task("inference")
 
     true_labels = test_dataset["label"]
     
@@ -160,7 +161,8 @@ async def evaluate_text(request: TextEvaluationRequest):
             predictions.extend(batch_preds)
     
     # Stop tracking emissions
-    emissions_data = tracker.stop_task()
+    emissions_data = stop_tracking()
+#    emissions_data = tracker.stop_task()
     
     # Calculate accuracy
     accuracy = accuracy_score(true_labels, predictions)

tasks/utils/emissions.py

@@ -1,17 +1,34 @@
 from codecarbon import EmissionsTracker
 import os
 
-# Initialize tracker
-tracker = EmissionsTracker(allow_multiple_runs=True)
+# Initialize tracker with basic configuration
+tracker = EmissionsTracker(
+    project_name="climate-guard",
+    output_dir=os.getenv("OUTPUT_DIR", "./"),
+    log_level='error'
+)
 
 class EmissionsData:
     def __init__(self, energy_consumed: float, emissions: float):
         self.energy_consumed = energy_consumed
         self.emissions = emissions
+        self.timestamp = None
+        self.project_name = None
+        self.experiment_id = None
+        self.latitude = None
+        self.longitude = None
 
 def clean_emissions_data(emissions_data):
     """Remove unwanted fields from emissions data"""
-    data_dict = emissions_data.__dict__
+    if hasattr(emissions_data, '__dict__'):
+        data_dict = emissions_data.__dict__
+    else:
+        # If emissions_data is not an object with __dict__
+        data_dict = {
+            'energy_consumed': getattr(emissions_data, 'energy_consumed', 0),
+            'emissions': getattr(emissions_data, 'emissions', 0)
+        }
+    
     fields_to_remove = ['timestamp', 'project_name', 'experiment_id', 'latitude', 'longitude']
     return {k: v for k, v in data_dict.items() if k not in fields_to_remove}
 
@@ -25,4 +42,27 @@ def get_space_info():
             return username, space_url
         except Exception as e:
             print(f"Error getting space info: {e}")
-    return "local-user", "local-development" 
+    return "local-user", "local-development"
+
+def start_tracking():
+    """Safely start the emissions tracking"""
+    try:
+        if not tracker._tracking:
+            tracker.start()
+        return True
+    except Exception as e:
+        print(f"Error starting emissions tracking: {e}")
+        return False
+
+def stop_tracking():
+    """Safely stop the emissions tracking and return data"""
+    try:
+        if tracker._tracking:
+            emissions = tracker.stop()
+            return EmissionsData(
+                energy_consumed=getattr(emissions, 'energy_consumed', 0),
+                emissions=getattr(emissions, 'emissions', 0)
+            )
+    except Exception as e:
+        print(f"Error stopping emissions tracking: {e}")
+        return EmissionsData(energy_consumed=0, emissions=0)