refactor: update royalty_shares type in ExtraData and clean up imports in registration_utils and test files

Author: bonnie57

src/story_protocol_python_sdk/types/utils.py

@@ -4,8 +4,6 @@
 from ens.ens import Address, HexStr
 from typing_extensions import Callable
 
-from story_protocol_python_sdk import RoyaltyShareInput
-
 
 class Multicall3Call(TypedDict):
     target: Address
@@ -38,7 +36,7 @@ class ExtraData(TypedDict, total=False):
         license_terms_data: [Optional] The license terms data.
     """
 
-    royalty_shares: list[RoyaltyShareInput]
+    royalty_shares: list[dict]
     deadline: int
     royalty_total_amount: int
     nft_contract: Address

src/story_protocol_python_sdk/utils/registration/registration_utils.py

@@ -5,12 +5,12 @@
 from web3 import Web3
 
 from story_protocol_python_sdk.abi.Multicall3.Multicall3_client import Multicall3Client
-from story_protocol_python_sdk.types.resource.IPAsset import (
+from story_protocol_python_sdk.types.resource.IPAsset import IPRoyaltyVault
+from story_protocol_python_sdk.types.utils import (
+    AggregatedRequestData,
     ExtraData,
-    IPRoyaltyVault,
     TransformedRegistrationRequest,
 )
-from story_protocol_python_sdk.types.utils import AggregatedRequestData
 from story_protocol_python_sdk.utils.registration.transform_registration_request import (
     transform_distribute_royalty_tokens_request,
 )

tests/unit/resources/test_ip_asset.py

@@ -24,10 +24,8 @@
     IPAccountImplClient,
 )
 from story_protocol_python_sdk.resources.IPAsset import IPAsset
-from story_protocol_python_sdk.types.resource.IPAsset import (
-    BatchMintAndRegisterIPInput,
-    TransformedRegistrationRequest,
-)
+from story_protocol_python_sdk.types.resource.IPAsset import BatchMintAndRegisterIPInput
+from story_protocol_python_sdk.types.utils import TransformedRegistrationRequest
 from story_protocol_python_sdk.utils.derivative_data import DerivativeDataInput
 from story_protocol_python_sdk.utils.ip_metadata import IPMetadata, IPMetadataInput
 from story_protocol_python_sdk.utils.royalty import get_royalty_shares

tests/unit/utils/test_registration_utils.py

@@ -4,9 +4,9 @@
 from ens.ens import HexStr
 
 from story_protocol_python_sdk import RoyaltyShareInput
-from story_protocol_python_sdk.types.resource.IPAsset import (
+from story_protocol_python_sdk.types.resource.IPAsset import IPRoyaltyVault
+from story_protocol_python_sdk.types.utils import (
     ExtraData,
-    IPRoyaltyVault,
     TransformedRegistrationRequest,
 )
 from story_protocol_python_sdk.utils.registration.registration_utils import (