Adding derive for FromPrimitive to enums.

metaplex-foundation · Apr 10, 2024 · 5b42445 · 5b42445
1 parent 3726e1f
commit 5b42445
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/renderers/rust/getTypeManifestVisitor.ts b/src/renderers/rust/getTypeManifestVisitor.ts
@@ -80,7 +80,10 @@ export function getTypeManifestVisitor() {
             isNode(definedType.type, 'enumTypeNode') &&
             isScalarEnum(definedType.type)
           ) {
-            traits.push('PartialOrd', 'Hash');
+            traits.push('PartialOrd', 'Hash', 'FromPrimitive');
+            manifest.imports.add([
+              'num_derive::FromPrimitive',
+            ]);
           }
           return {
             ...manifest,