Merge branch 'master' into fix-frontent

gnes-ai · Oct 15, 2019 · 40fa503 · 40fa503
2 parents 474dedd + 2d398c1
commit 40fa503
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)))