Fix: Load distribution and CPU calculation

env/core.py

@@ -55,12 +55,12 @@ TASKS = {
         difficulty="medium",
         description="Performance bottleneck! This cluster is struggling. Analyze each server's load, then iteratively upgrade undersized servers. Requires 4+ successful changes for max reward.",
         initial_resources=[
-            {"id": "srv-1", "type": "t3.small", "cpu_usage": 60.0, "mem_usage": 50.0, "monthly_cost": 11.5},
-            {"id": "srv-2", "type": "t3.small", "cpu_usage": 55.0, "mem_usage": 45.0, "monthly_cost": 11.5},
-            {"id": "srv-3", "type": "t3.small", "cpu_usage": 58.0, "mem_usage": 48.0, "monthly_cost": 11.5},
+            {"id": "srv-1", "type": "t3.small", "cpu_usage": 40.0, "mem_usage": 30.0, "monthly_cost": 11.5},
+            {"id": "srv-2", "type": "t3.small", "cpu_usage": 38.0, "mem_usage": 28.0, "monthly_cost": 11.5},
+            {"id": "srv-3", "type": "t3.small", "cpu_usage": 42.0, "mem_usage": 32.0, "monthly_cost": 11.5},
         ],
-        sla={"max_latency_ms": 100.0, "max_budget": 100.0, "min_uptime_pct": 99.9},
-        load=10.0
+        sla={"max_latency_ms": 100.0, "max_budget": 80.0, "min_uptime_pct": 99.9},
+        load=4.5
     ),
     "hard": TaskConfig(
         task_id="hard_balance",
@@ -70,12 +70,12 @@ TASKS = {
         initial_resources=[
             {"id": "srv-1", "type": "m5.large", "cpu_usage": 15.0, "mem_usage": 10.0, "monthly_cost": 70.0},
             {"id": "srv-2", "type": "m5.large", "cpu_usage": 12.0, "mem_usage": 8.0, "monthly_cost": 70.0},
-            {"id": "srv-3", "type": "t3.small", "cpu_usage": 70.0, "mem_usage": 60.0, "monthly_cost": 11.5},
-            {"id": "srv-4", "type": "t3.small", "cpu_usage": 75.0, "mem_usage": 65.0, "monthly_cost": 11.5},
-            {"id": "srv-5", "type": "t3.medium", "cpu_usage": 45.0, "mem_usage": 40.0, "monthly_cost": 23.0},
+            {"id": "srv-3", "type": "t3.small", "cpu_usage": 50.0, "mem_usage": 40.0, "monthly_cost": 11.5},
+            {"id": "srv-4", "type": "t3.small", "cpu_usage": 55.0, "mem_usage": 45.0, "monthly_cost": 11.5},
+            {"id": "srv-5", "type": "t3.medium", "cpu_usage": 35.0, "mem_usage": 30.0, "monthly_cost": 23.0},
         ],
-        sla={"max_latency_ms": 100.0, "max_budget": 70.0, "min_uptime_pct": 99.9},
-        load=25.0
+        sla={"max_latency_ms": 100.0, "max_budget": 80.0, "min_uptime_pct": 99.9},
+        load=15.0
     ),
 }
 
@@ -279,8 +279,11 @@ class CloudOpsEnvironment:
             self._ep.resources
         )
         
+        total_capacity = sum(INSTANCE_DATA[r.type]["capacity"] for r in self._ep.resources)
+        
         for r in self._ep.resources:
-            r.cpu_usage = min(100.0, self._ep.current_load / INSTANCE_DATA[r.type]["capacity"] * 100)
+            cap = INSTANCE_DATA[r.type]["capacity"]
+            r.cpu_usage = min(100.0, self._ep.current_load / total_capacity / cap * 100)
             r.mem_usage = min(100.0, r.cpu_usage * 0.9)
         
         metrics = Metrics(