diff --git a/pricing/init.py b/pricing/init.py index bed603d..425f119 100644 --- a/pricing/init.py +++ b/pricing/init.py @@ -1,7 +1,6 @@ from pricing.pricing import ( PricingProgram, test_pricing, - buffered_charging, get_pricing_program ) from ahserver.serverenv import ServerEnv @@ -9,9 +8,9 @@ from ahserver.serverenv import ServerEnv def load_pricing(): env = ServerEnv() env.get_pricing_program = get_pricing_program - env.buffered_charging = buffered_charging env.write_pricing_patten = PricingProgram.write_pricing_patten env.pricing_program_charging = PricingProgram.charging + env.buffered_charging = PricingProgram.buffered_charging env.load_pricing_data = PricingProgram.load_pricing_data env.get_pricing_program = PricingProgram.get_pricing_program env.test_pricing = test_pricing