simple code cleanup

This commit is contained in:
2025-12-17 18:50:04 +01:00
parent c8c44d0453
commit 57a7e0c571
2 changed files with 8 additions and 1 deletions

View File

@@ -39,7 +39,7 @@ class EngineTrainer:
obs, _ = self.env.reset(seed=seed)
prices = None
for ep in range(n_episodes):
prices = self.engine.compute_prices(prices, obs
prices = self.engine.compute_prices(prices, obs)
obs, reward, done, _, info = self.env.step(prices)
self.engine.update(obs, reward, done, info)
return self