From aa21e8aa91fe893b5c460200b74c928ed918e28a Mon Sep 17 00:00:00 2001
From: Xuesong Yang <1646669+XuesongYang@users.noreply.github.com>
Date: Mon, 5 Jun 2023 11:44:09 -0700
Subject: [PATCH] fixed  decor to show messages only when the wrapped object is
 called. (#6793)

Signed-off-by: Xuesong Yang <1646669+XuesongYang@users.noreply.github.com>
---
 nemo/utils/decorators/experimental.py | 18 ++++++------------
 1 file changed, 6 insertions(+), 12 deletions(-)

diff --git a/nemo/utils/decorators/experimental.py b/nemo/utils/decorators/experimental.py
index 35b26fb8690d..de62dbaf9ffb 100644
--- a/nemo/utils/decorators/experimental.py
+++ b/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)