diff options
| -rw-r--r-- | candle-core/src/pickle.rs | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/candle-core/src/pickle.rs b/candle-core/src/pickle.rs index f6d80830..e3f1f81d 100644 --- a/candle-core/src/pickle.rs +++ b/candle-core/src/pickle.rs @@ -350,8 +350,10 @@ impl Stack { module_name, class_name, } => { - if module_name == "collections" && class_name == "OrderedDict" { - // TODO: have a separate ordered dict. + if module_name == "collections" + && (class_name == "OrderedDict" || class_name == "defaultdict") + { + // TODO: have a separate ordered dict and a separate default dict. Some(Object::Dict(vec![])) } else { None |