Refactor numpy serializer

Signed-off-by: Emanuele Giaquinta <[email protected]>
aviramha · Dec 28, 2024 · 531cc95 · 531cc95
1 parent a98466d
commit 531cc95
Show file tree

Hide file tree

Showing 3 changed files with 366 additions and 635 deletions.
diff --git a/src/serialize/dict.rs b/src/serialize/dict.rs
@@ -252,16 +252,23 @@ impl Serialize for DictWithNonStrKeys {
                 )?;
             } else {
                 match pyobject_to_obtype(key.as_ptr(), opts) {
-                    ObType::NumpyScalar
-                    | ObType::NumpyArray
-                    | ObType::Dict
-                    | ObType::List
-                    | ObType::Dataclass
-                    | ObType::Pydantic
-                    | ObType::Unknown => {
+                    ObType::Bool
+                    | ObType::Bytes
+                    | ObType::Date
+                    | ObType::Datetime
+                    | ObType::Enum
+                    | ObType::Ext
+                    | ObType::Float
+                    | ObType::Int
+                    | ObType::None
+                    | ObType::Str
+                    | ObType::StrSubclass
+                    | ObType::Time
+                    | ObType::Tuple
+                    | ObType::Uuid => (),
+                    _ => {
                         err!("Dict key must a type serializable with OPT_NON_STR_KEYS")
                     }
-                    _ => (),
                 }
                 map.serialize_entry(
                     &PyObject::new(