File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -1382,7 +1382,6 @@ def get_nbytes(obj):
13821382 and not i .is_ArrayBasic and not i .alias ]
13831383
13841384 for i in op_symbols :
1385- # Will overreport memory usage currently
13861385 try :
13871386 # TODO: is _obj even needed?
13881387 v = get_nbytes (self [i .name ]._obj )
@@ -1467,7 +1466,6 @@ def nbytes_consumed_memmapped(self):
14671466
14681467 @cached_property
14691468 def nbytes_snapshots (self ):
1470-
14711469 # Symbols in the operator which may or may not carry data
14721470 op_symbols = FindSymbols ().visit (self .op )
14731471
@@ -1478,7 +1476,12 @@ def nbytes_snapshots(self):
14781476 disk = 0
14791477 for i in op_symbols :
14801478 try :
1481- disk += i .size_snapshot * i ._time_size_ideal * np .dtype (i .dtype ).itemsize
1479+ v = self [i .name ]._obj
1480+ except AttributeError :
1481+ v = self .get (i .name , i )
1482+
1483+ try :
1484+ disk += v .size_snapshot * v ._time_size_ideal * np .dtype (v .dtype ).itemsize
14821485 except AttributeError :
14831486 pass
14841487
You can’t perform that action at this time.
0 commit comments