From e22286371f8ee5b1a0fdcdc957e6545d3b339162 Mon Sep 17 00:00:00 2001
From: Daniel Rosel <daniel@alves.world>
Date: Fri, 6 Feb 2026 11:54:23 +0100
Subject: [PATCH] feat: proportiona lrevenu

---
 engine/wrapper.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/engine/wrapper.py b/engine/wrapper.py
index 52ee5b6..fe1e6bb 100644
--- a/engine/wrapper.py
+++ b/engine/wrapper.py
@@ -135,7 +135,8 @@ class PHANTOM(gym.Env):
         self.market.alpha, self.market.Nagents, self.market.Nhumans = old_state
 
         coi_leakage = max(0.0, coi_base - coi_mix)
-        coi_penalty = self.lambda_coi * coi_leakage
+        coi_penalty = max(self.lambda_coi * coi_leakage, 1000) / 1000
+        coi_penalty *= revenue
 
         return float(revenue - coi_penalty), {
             "revenue": float(revenue),