Updated imports to be relative in market module, fix return type on get method

atbe · atbe · commit 33db7099e189 · 2021-11-09T13:33:40.000-08:00
diff --git a/thirdweb/modules/market.py b/thirdweb/modules/market.py
@@ -1,17 +1,16 @@
-from typing import Dict, List
+from typing import List
 
 from thirdweb_web3 import Web3
 
-from nftlabs.modules.currency import CurrencyModule
-from nftlabs.modules.nft import NftModule
-
 from ..abi.erc20 import ERC20
 from ..abi.erc1155 import ERC1155
 from ..abi.market import Market, MarketListing
 from ..abi.nft import NFT
 from ..constants import ZeroAddress
 from ..errors import AssetNotFoundException, UnsupportedAssetException
-from ..types.currency import Currency, CurrencyValue
+from ..modules.currency import CurrencyModule
+from ..modules.nft import NftModule
+from ..types.currency import CurrencyValue
 from ..types.listing import Listing
 from ..types.market import Filter, ListArg, MarketListing
 from . import BaseModule
@@ -148,7 +147,7 @@ def set_market_fee_bps(self, amount: int):
             self.get_transact_opts())
         self.execute_tx(tx)
 
-    def get(self, listing_id) -> MarketListing:
+    def get(self, listing_id) -> Listing:
         """
         Get a listing.
         """
