fixed signatures of decorated functions

neural_monitor/monitor.py

@@ -17,7 +17,6 @@
 from imageio import imwrite
 from shutil import copyfile
 from collections import namedtuple, deque
-import functools
 import torch.distributed
 from typing import (
     List,
@@ -30,6 +29,7 @@
 )
 import pandas as pd
 from easydict import EasyDict
+from decorator import decorator
 
 try:
     import visdom
@@ -184,8 +184,8 @@ def _spawn_defaultdict_ordereddict():
     return collections.OrderedDict()
 
 
+@decorator
 def check_path_init(f):
-    @functools.wraps(f)
     def set_default_path(self, *args, **kwargs):
         if not self._initialized:
             logger.info('Working folder not initialized! Initialize working folder to default.')
@@ -197,19 +197,19 @@ def set_default_path(self, *args, **kwargs):
     return set_default_path
 
 
+@decorator
 def standardize_name(f):
-    @functools.wraps(f)
     def func(self, name: str, *args, **kwargs):
         if name is not None:
             name = name.replace(' ', '-')
 
-        f(self, name, *args, **kwargs)
+        return f(self, name, *args, **kwargs)
 
     return func
 
 
+@decorator
 def distributed_collect(f):
-    @functools.wraps(f)
     def func(self, name: str, value: T.Tensor, *args, **kwargs):
         if self.distributed:
             assert isinstance(value, T.Tensor), 'value must be a Tensor in distributed mode'
@@ -220,8 +220,8 @@ def func(self, name: str, value: T.Tensor, *args, **kwargs):
     return func
 
 
+@decorator
 def distributed_flush(f):
-    @functools.wraps(f)
     def func(self, *args, **kwargs):
         if self.distributed and self.rank != 0:
             return

requirements.txt

@@ -8,3 +8,5 @@ tensorboard
 imageio
 git-python
 pandas
+decorator
+easydict

setup.py

@@ -30,8 +30,9 @@ def setup_package():
             'Programming Language :: Python :: 3.7'
         ],
         platforms=['Windows', 'Linux'],
-        packages=find_packages(exclude=['docs']),
-        install_requires=['matplotlib', 'numpy', 'imageio', 'tensorboard', 'git-python', 'easydict', 'pandas'],
+        packages=find_packages(exclude=['docs', 'doc']),
+        install_requires=['matplotlib', 'numpy', 'imageio', 'tensorboard',
+                          'git-python', 'easydict', 'pandas', 'decorator'],
         project_urls={
             'Bug Reports': 'https://github.com/justanhduc/neural-monitor/issues',
             'Source': 'https://github.com/justanhduc/neural-monitor',