Mirror types instead of replacements

atbe · atbe · commit c1ad584830c4 · 2021-11-11T13:15:52.000-08:00
diff --git a/thirdweb/modules/__init__.py b/thirdweb/modules/__init__.py
@@ -5,5 +5,5 @@
 from .currency import *
 from .market import *
 from .pack import *
-from .collection import *
+from .collection import CollectionModule
 from .bundle import *
diff --git a/thirdweb/modules/collection.py b/thirdweb/modules/collection.py
diff --git a/thirdweb/modules/collection/__init__.py b/thirdweb/modules/collection/__init__.py
@@ -0,0 +1,3 @@
+from ..bundle import BundleModule
+
+CollectionModule = BundleModule
diff --git a/thirdweb/types/collection/__init__.py b/thirdweb/types/collection/__init__.py
@@ -1,40 +1 @@
-from dataclasses import dataclass
-from typing import Optional
-
-from dataclasses_json import dataclass_json
-
-from ..nft import NftMetadata
-from ..metadata import Metadata
-
-
-@dataclass_json
-@dataclass
-class CollectionMetadata:
-    """
-    COLLECTION MODULE AND ALL ITS CLASSES WILL BE DEPRECATED SOON. USE BUNDLE MODULE INSTEAD.
-    """
-    id: Optional[int] = None
-    creator: Optional[str] = None
-    supply: Optional[int] = None
-    metadata: Optional[NftMetadata] = None
-
-
-@dataclass_json
-@dataclass
-class CreateCollectionArg:
-    """
-    COLLECTION MODULE AND ALL ITS CLASSES WILL BE DEPRECATED SOON. USE BUNDLE MODULE INSTEAD.
-    """
-    metadata: Optional[Metadata] = None
-    supply: Optional[int] = None
-
-
-@dataclass_json
-@dataclass
-class MintCollectionArg:
-    """
-    COLLECTION MODULE AND ALL ITS CLASSES WILL BE DEPRECATED SOON. USE BUNDLE MODULE INSTEAD.
-    """
-    token_id: Optional[int] = None
-    amount: Optional[int] = None
-
+from .types import CreateCollectionArg, CollectionMetadata, MintCollectionArg

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+from ..bundle import BundleModule`
	`2`	`+`
	`3`	`+CollectionModule = BundleModule`