fixed decor to show messages only when the wrapped object is called. (N…

…VIDIA#6793)

Signed-off-by: Xuesong Yang <1646669+XuesongYang@users.noreply.github.com>

nemo/utils/decorators/experimental.py

@@ -15,19 +15,13 @@
 
 __all__ = ['experimental']
 
-from nemo.utils import logging
-
 
-def experimental(cls):
- """ Decorator which indicates that module is experimental.
- Use it to mark experimental or research modules.
- """
+import wrapt
 
- def wrapped(cls):
- logging.warning(
- f'Module {cls} is experimental, not ready for production and is not fully supported. Use at your own risk.'
- )
+from nemo.utils import logging
 
- return cls
 
- return wrapped(cls=cls)
+@wrapt.decorator
+def experimental(wrapped, instance, args, kwargs):
+ logging.warning(f"`{wrapped}` is experimental and not ready for production yet. Use at your own risk.")
+ return wrapped(*args, **kwargs)