Merge pull request #894 from japm48/patch-1

npy: fix parsing of complex dtypes
LaurentMazare · Sep 15, 2024 · 1fd7075 · 1fd7075
2 parents 9c498e6 + 5758c51
commit 1fd7075
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/tensor/npy.rs b/src/tensor/npy.rs
@@ -140,8 +140,8 @@ impl Header {
                     "b" | "i1" => Kind::Int8,
                     "B" | "u1" => Kind::Uint8,
                     "?" | "b1" => Kind::Bool,
-                    "F" | "F4" => Kind::ComplexFloat,
-                    "D" | "F8" => Kind::ComplexDouble,
+                    "F" | "F4" | "c8" => Kind::ComplexFloat,
+                    "D" | "F8" | "c16" => Kind::ComplexDouble,
                     descr => {
                         return Err(TchError::FileFormat(format!("unrecognized descr {descr}")))
                     }