drl-supply-chain-env-initialization.py

class SupplyChainEnvironment(object):
    def __init__(self):
        self.T = 26               # Episode duration
        self.warehouse_num = 3  
        self.d_max = 5            # Maximum demand, units
        self.d_var = 2            # Maximum random demand variation, units 
    
        self.unit_price = 100     # Unit price in dollars
        self.unit_cost = 40       # Unit cost in dollars

        self.storage_capacities = np.fromfunction(lambda j: 10*(j+1), 
                                  (self.warehouse_num + 1,))

        # Storage costs at the factory and each warehouse, dollars per unit
        self.storage_costs = np.fromfunction(lambda j: 2*(j+1), (self.warehouse_num + 1,))
        
        # Transportation costs for each warehouse, dollars per unit   
        self.transporation_costs = np.fromfunction(lambda j: 5*(j+1), (self.warehouse_num,))               
        
        self.penalty_unit_cost = self.unit_price