minor

ai4co · May 16, 2022 · 2a34396 · 2a34396
1 parent 4be4246
commit 2a34396
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/torchrl/__init__.py b/torchrl/__init__.py
@@ -49,11 +49,9 @@ def decorated_fn(*args, **kwargs):
         return decorated_fn
 
     def __enter__(self):
-        return
         self.t0 = time.time()
 
     def __exit__(self, exc_type, exc_val, exc_tb):
-        return
         t = time.time() - self.t0
         self._REG.setdefault(self.name, [0.0, 0.0, 0])
 
@@ -64,7 +62,6 @@ def __exit__(self, exc_type, exc_val, exc_tb):
 
     @staticmethod
     def print():
-        return
         keys = list(timeit._REG)
         keys.sort()
         for name in keys: