revert inference code

tasks/utils/emissions.py

@@ -1,13 +1,8 @@
-# tasks/utils/emissions.py
 from codecarbon import EmissionsTracker
 import os
 
-# Initialize tracker with correct parameters
-tracker = EmissionsTracker(
-    project_name="frugal-ai-challenge",
-    output_dir=".",
-    log_level='warning'
-)
+# Initialize tracker
+tracker = EmissionsTracker(allow_multiple_runs=True)
 
 class EmissionsData:
     def __init__(self, energy_consumed: float, emissions: float):
@@ -16,15 +11,8 @@ class EmissionsData:
 
 def clean_emissions_data(emissions_data):
     """Remove unwanted fields from emissions data"""
-    if isinstance(emissions_data, dict):
-        data_dict = emissions_data
-    else:
-        data_dict = emissions_data.__dict__
-    
-    fields_to_remove = [
-        'timestamp', 'project_name', 'experiment_id', 
-        'latitude', 'longitude', '_start_time', '_end_time'
-    ]
+    data_dict = emissions_data.__dict__
+    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}
 
 def get_space_info():
@@ -37,23 +25,4 @@ def get_space_info():
             return username, space_url
         except Exception as e:
             print(f"Error getting space info: {e}")
-    return "local-user", "local-development"
-
-def start_tracking():
-    """Start tracking emissions"""
-    try:
-        tracker.start()
-    except Exception as e:
-        print(f"Error starting emissions tracking: {e}")
-
-def stop_tracking():
-    """Stop tracking emissions and return data"""
-    try:
-        emissions = tracker.stop()
-        return EmissionsData(
-            energy_consumed=emissions.energy_consumed,
-            emissions=emissions.emissions
-        )
-    except Exception as e:
-        print(f"Error stopping emissions tracking: {e}")
-        return EmissionsData(energy_consumed=0.0, emissions=0.0)
+    return "local-user", "local-development"