removed enable grad from closure path

Signed-off-by: mikail <mkhona@nvidia.com>

Author: mkhona-nvidia

emerging_optimizers/scalar_optimizers/laprop.py

@@ -144,10 +144,10 @@ def step(self, closure: Callable[[], float] | None = None) -> float | None:
         Returns:
             The loss from the closure, if provided.
         """
-        loss = None
-        if closure is not None:
-            with torch.enable_grad():
-                loss = closure()
+        if closure is None:
+            loss = None
+        else:
+            loss = closure()
 
         for group in self.param_groups:
             self._init_group(group)