From 47af7dfe6a7ed3f969f83dfea4223773c5bdf87c Mon Sep 17 00:00:00 2001 From: Miroslav Stampar Date: Wed, 7 Jan 2015 10:49:15 +0100 Subject: [PATCH] Another minor patch --- lib/core/bigarray.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/lib/core/bigarray.py b/lib/core/bigarray.py index d62588c21..faf499ba0 100644 --- a/lib/core/bigarray.py +++ b/lib/core/bigarray.py @@ -10,6 +10,7 @@ try: except: import pickle +import itertools import os import sys import tempfile @@ -24,7 +25,9 @@ def _size_of(object): """ retval = sys.getsizeof(object) - if hasattr(object, "__iter__"): + if isinstance(object, dict): + retval += sum(_size_of(_) for _ in itertools.chain.from_iterable(object.items())) + elif hasattr(object, "__iter__"): retval += sum(_size_of(_) for _ in object) return retval