diff --git a/modules/deeplearning/icing.py b/modules/deeplearning/icing.py index 7b1a8dec874338525583d97f6bd0e7673ce68671..cb8b00e8ba1c37cbd7563ef44b2a0fa59310fec1 100644 --- a/modules/deeplearning/icing.py +++ b/modules/deeplearning/icing.py @@ -182,9 +182,9 @@ class IcingIntensityNN: key = frozenset(idxs) if CACHE_DATA_IN_MEM: - data, label = self.in_mem_data_cache.get(key) - if data is not None: - return data, label + tup = self.in_mem_data_cache.get(key) + if tup is not None: + return tup[0], tup[1] # sort these to use as numpy indexing arrays nd_idxs = np.array(idxs)