Merge pull request #327 from gnes-ai/fix-doc-1

refactor(flow): flow is now a trainable base
gnes-ai · Oct 15, 2019 · 2d398c1 · 2d398c1
2 parents 0ac92e4 + 166698c
commit 2d398c1
Show file tree

Hide file tree

Showing 5 changed files with 261 additions and 172 deletions.
diff --git a/gnes/base/__init__.py b/gnes/base/__init__.py
@@ -107,7 +107,7 @@ def register_class(cls):
             # print('reg class: %s' % cls.__name__)
             cls.__init__ = TrainableType._store_init_kwargs(cls.__init__)
             if os.environ.get('GNES_PROFILING', False):
-                for f_name in ['train', 'encode', 'add', 'query']:
+                for f_name in ['train', 'encode', 'add', 'query', 'index']:
                     if getattr(cls, f_name, None):
                         setattr(cls, f_name, profiling(getattr(cls, f_name)))