diff options
Diffstat (limited to 'tensorflow/python/client/session.py')
-rw-r--r-- | tensorflow/python/client/session.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/client/session.py b/tensorflow/python/client/session.py index f3b788f931..e037925961 100644 --- a/tensorflow/python/client/session.py +++ b/tensorflow/python/client/session.py @@ -361,7 +361,7 @@ class _ListFetchMapper(_FetchMapper): for m, vi in zip(self._mappers, self._value_indices): results.append(m.build_results([values[j] for j in vi])) # Return a value of the original type of the fetches. - if self._fetch_type == list: + if issubclass(self._fetch_type, list): return results elif self._fetch_type == tuple: return tuple(results) |