Add ODS Logging to Captum (pytorch#971)

Summary: Pull Request resolved: pytorch#971

Reviewed By: NarineK

Differential Revision: D37009629

fbshipit-source-id: 161a957ed56abfb734c9004fc8420e66ccde9d20

Author: vivekmig

captum/log/__init__.py

@@ -2,14 +2,21 @@
 
 try:
     from captum.log.fb.internal_log import (
+        disable_detailed_logging,
         log,
         log_usage,
         patch_methods,
         set_environment,
         TimedLog,
     )
 
-    __all__ = ["log", "log_usage", "TimedLog", "set_environment"]
+    __all__ = [
+        "log",
+        "log_usage",
+        "TimedLog",
+        "set_environment",
+        "disable_detailed_logging",
+    ]
 
 except ImportError:
     from functools import wraps
@@ -41,5 +48,8 @@ def wrapper(*args, **kwargs):
     def set_environment(env):
         pass
 
+    def disable_detailed_logging():
+        pass
+
     def patch_methods(tester, patch_log=True):
         pass