luxolus/pyo3
2
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

dict: use HashMap::with_capacity in extract

Browse source
This commit is contained in:
David Hewitt 2020-11-26 10:01:14 +00:00
parent 02b90c462f
commit 1b0642fdd5
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/types/dict.rs
View file

@ -331,7 +331,7 @@ where
{ {
fn extract(ob: &'source PyAny) -> Result<Self, PyErr> { fn extract(ob: &'source PyAny) -> Result<Self, PyErr> {
let dict = <PyDict as PyTryFrom>::try_from(ob)?; let dict = <PyDict as PyTryFrom>::try_from(ob)?;
let mut ret = HashMap::default(); let mut ret = HashMap::with_capacity_and_hasher(dict.len(), S::default());
for (k, v) in dict.iter() { for (k, v) in dict.iter() {
ret.insert(K::extract(k)?, V::extract(v)?); ret.insert(K::extract(k)?, V::extract(v)?);
} }
@ -392,7 +392,7 @@ mod hashbrown_hashmap_conversion {
{ {
fn extract(ob: &'source PyAny) -> Result<Self, PyErr> { fn extract(ob: &'source PyAny) -> Result<Self, PyErr> {
let dict = <PyDict as PyTryFrom>::try_from(ob)?; let dict = <PyDict as PyTryFrom>::try_from(ob)?;
let mut ret = hashbrown::HashMap::default(); let mut ret = hashbrown::HashMap::with_capacity_and_hasher(dict.len(), S::default());
for (k, v) in dict.iter() { for (k, v) in dict.iter() {
ret.insert(K::extract(k)?, V::extract(v)?); ret.insert(K::extract(k)?, V::extract(v)?);
} }