update

tensorpack/dataflow/raw.py

@@ -111,6 +111,9 @@ class DataFromGenerator(DataFlow):
         else:
             yield from self._gen()
 
+    def __len__(self):
+        return len(self._gen)
+
 
 class DataFromIterable(DataFlow):
     """ Wrap an iterable of datapoints to a DataFlow"""

tensorpack/tfutils/sessinit.py

@@ -260,7 +260,7 @@ class ChainInit(SessionInit):
             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.