fixed all environment files

demand_calculator.py

@@ -4,7 +4,7 @@ from dataclasses import dataclass
 from typing import Tuple
 from config import LEAD_TIME
 
-
+#Demand_Calculator
 np.random.seed(42)
 
 @dataclass

demand_environment.py

@@ -5,7 +5,7 @@ from config import SHAPE_GAMMA_POISSON, LAMBDA_GAMMA_POISSON, SCALE_GAMMA_POISSO
     SCALE_GAMMA_GAMMA_LOW_MEAN, SHAPE_GAMMA_GAMMA_HIGH_MEAN, SCALE_GAMMA_GAMMA_HIGH_MEAN, SHAPE_GAMMA_LOW_VAR, \
     SCALE_GAMMA_LOW_VAR, RATE_SPORADIC_HIGH
 from demand_calculator import DailyDemandDistribution, DemandCalculator
-
+#Demand_Environment
 class Environment(ABC):
     def __init__(self, sim_days: int):
         self.sim_days = sim_days

inventory_manager.py

@@ -1,5 +1,5 @@
 from config import WRITE_OFF_RATE, BASE_STOCK, DEFAULT_SERVICE_LEVEL, MC_SIMS
-
+#Inventory_Manager
 class InventoryManager:
     def __init__(self, order_processor, agent):
         self.inventory = BASE_STOCK

main.py

@@ -5,7 +5,7 @@ from demand_calculator import DemandCalculator
 from config import SIM_DAYS, N_SIMULATIONS
 import ciw
 ciw.seed(11)
-
+#Main
 simulation_version = input("Choose a simulation version (0-3):\n"
                          "0: Historical Demand Agent\n"
                          "1: Safety Stock Agent\n" 

montecarlo_simulator.py

@@ -4,7 +4,7 @@ from demand_calculator import DemandCalculator
 from order_processor import OrderProcessor
 from inventory_manager import InventoryManager
 from performance_tracker import PerformanceTracker
-
+#MC_Simulator
 class MonteCarloSimulator():
     def __init__(self, agent, environment):
         self.agent = agent

order_processor.py

@@ -2,7 +2,7 @@ from dataclasses import dataclass
 from typing import List
 import numpy as np
 from config import LEAD_TIME, LEAD_TIME_JITTER
-
+#Order_processor
 @dataclass
 class Order:
     arrival_day: int

performance_tracker.py

@@ -1,3 +1,4 @@
+#KPIs
 class PerformanceTracker:
 
     def __init__(self):