proper init/import

Author: 0xbigz

src/driftpy/math/__init__.py

@@ -1,6 +1,4 @@
 from driftpy.constants.numeric_constants import MARK_PRICE_PRECISION, PEG_PRECISION
-
-
 from driftpy.types import SwapDirection, AssetType, PositionDirection
 
 

src/driftpy/math/amm.py

@@ -1,4 +1,4 @@
-from amm import calculate_price
+from driftpy.math.amm import calculate_price
 
 
 def calculate_mark_price(market):

src/driftpy/math/market.py

@@ -1,6 +1,7 @@
-from market import calculate_mark_price
-from amm import calculate_price, calculate_amm_reserves_after_swap
 import math
+from driftpy.math.amm import calculate_price, calculate_amm_reserves_after_swap
+from driftpy.math.market import calculate_mark_price
+
 from driftpy.constants import (
     MARK_PRICE_PRECISION,
     PEG_PRECISION,