update

torch_geometric/debug.py

@@ -34,15 +34,16 @@ class set_debug:
 
     :class:`set_debug` will enable or disable the debug mode based on its
     argument :attr:`mode`.
+    It can be used as a context-manager or as a function.
 
     See :class:`debug` above for more details.
     """
     def __init__(self, mode: bool):
         self.prev = is_debug_enabled()
-        self.mode = mode
+        set_debug_enabled(mode)
 
     def __enter__(self):
-        set_debug_enabled(self.mode)
+        pass
 
     def __exit__(self, *args):
         set_debug_enabled(self.prev)

torch_geometric/experimental.py

@@ -62,19 +62,20 @@ class set_experimental_mode:
 
     :class:`set_experimental_mode` will enable or disable the experimental mode
     based on its argument :attr:`mode`.
+    It can be used as a context-manager or as a function.
 
     See :class:`experimental_mode` above for more details.
     """
     def __init__(self, mode: bool, options: Options = None):
-        self.mode = mode
         self.options = get_options(options)
         self.previous_state = {
             option: __experimental_flag__[option]
             for option in self.options
         }
+        set_experimental_mode_enabled(mode, self.options)
 
     def __enter__(self):
-        set_experimental_mode_enabled(self.mode, self.options)
+        pass
 
     def __exit__(self, *args):
         for option, value in self.previous_state.items():