Fixing linting issues

pytorch · May 14, 2024 · 543bba1 · 543bba1
1 parent a0af2fe
commit 543bba1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/test/test_cuda.py b/test/test_cuda.py
@@ -4465,10 +4465,10 @@ def test_graph_optims(self, device, dtype, optim_info):
 
         for optim_input in all_optim_inputs:
             kwargs = optim_input.kwargs
-            
+
             # lr as a Tensor is not supported when capturable=False and foreach=True for torch.optim.adam
             # and torch.optim.adamw
-            kwargs["lr"] = 0.1 
+            kwargs["lr"] = 0.1
 
             for actually_do_graphs in (True, False):
                 params = [

diff --git a/torch/testing/_internal/common_optimizers.py b/torch/testing/_internal/common_optimizers.py
@@ -323,7 +323,6 @@ def optim_inputs_func_adadelta(device, dtype=None):
         OptimizerInput(
             params=None, kwargs={"rho": 0.95, "weight_decay": 0.9}, desc="rho"
         ),
-
     ] + (cuda_supported_configs if "cuda" in str(device) else [])
 
 
@@ -541,7 +540,6 @@ def optim_inputs_func_adamax(device, dtype=None):
             kwargs={"weight_decay": 0.1, "maximize": True},
             desc="maximize, weight_decay",
         ),
-
     ] + (cuda_supported_configs if "cuda" in str(device) else [])