fix imports

enthought · Mar 24, 2017 · f6347c7 · f6347c7
1 parent 0c106dc
commit f6347c7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/quantlib/models/api.py b/quantlib/models/api.py
@@ -1,4 +1,5 @@
-from .hestonmodel import HestonModelHelper, HestonMode
-from .batesmodel import (BatesModel, BatesDetJumpModel, BatesDoubleExpModel,
+from .equity.heston_model import HestonModelHelper, HestonModel
+from .equity.bates_model import (BatesModel, BatesDetJumpModel, BatesDoubleExpModel,
     BatesDoubleExpDetJumpModel)
 from .calibration_helper import (RelativePriceError, PriceError, ImpliedVolError)
+from .shortrate.onefactormodels.hullwhite import HullWHite