update

tensorpack · Mar 17, 2020 · 22582cc · 22582cc
1 parent bc5b5cc
commit 22582cc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/tensorpack/dataflow/raw.py b/tensorpack/dataflow/raw.py
@@ -111,6 +111,9 @@ def __iter__(self):
         else:
             yield from self._gen()
 
+    def __len__(self):
+        return len(self._gen)
+
 
 class DataFromIterable(DataFlow):
     """ Wrap an iterable of datapoints to a DataFlow"""

diff --git a/tensorpack/tfutils/sessinit.py b/tensorpack/tfutils/sessinit.py
@@ -260,7 +260,7 @@ def _run_init(self, sess):
             i._run_init(sess)
 
 
-def SmartInit(obj, ignore_mismatch=False):
+def SmartInit(obj, *, ignore_mismatch=False):
     """
     Create a :class:`SessionInit` to be loaded to a session,
     automatically from any supported objects, with some smart heuristics.