non_linearities.py

# adding reward for moving cars from one location to another (which is negative) 
    
if action <= 0:
    ψ = ψ + jcp.moving_reward() * abs(action)
else:
    ψ = ψ + jcp.moving_reward() * (action - 1)    #one car is moved by one of Jack's employees for free

# adding reward for second parking lot (which is also negative)

if new_state[0] > 10:
    ψ = ψ + jcp.second_parking_lot_reward()

if new_state[1] > 10:
    ψ = ψ + jcp.second_parking_lot_reward()