From 09744fe2fee790365167827e2f589cb333fb50d5 Mon Sep 17 00:00:00 2001
From: Yukio Siraichi <yukio.siraichi@gmail.com>
Date: Thu, 18 Jan 2024 17:56:10 -0300
Subject: [PATCH] Fix lint issues.

---
 benchmarks/torchbench_model.py | 32 ++++++++++++++++----------------
 1 file changed, 16 insertions(+), 16 deletions(-)

diff --git a/benchmarks/torchbench_model.py b/benchmarks/torchbench_model.py
index b4600871bc6..d715ee6dab0 100644
--- a/benchmarks/torchbench_model.py
+++ b/benchmarks/torchbench_model.py
@@ -32,22 +32,22 @@
 # torchbench models that might OOM using Adam.
 # This list was extracted from PyTorch's repository: benchmarks/dynamo/common.py
 TRAIN_WITH_SGD = {
-  "torchrec_dlrm",
-  "demucs",
-  "hf_T5_base",
-  "hf_clip",
-  "llama_v2_7b_16h",
-  "mobilenet_v2_quantized_qat",
-  "phi_1_5",
-  "resnet50_quantized_qat",
-  "DALLE2_pytorch",
-  "moco",
-  "timm_efficientdet",
-  "pytorch_CycleGAN_and_pix2pix",
-  "vision_maskrcnn",
-  "dlrm",
-  "resnet50",
-  *DETECTRON2_MODELS,
+    "torchrec_dlrm",
+    "demucs",
+    "hf_T5_base",
+    "hf_clip",
+    "llama_v2_7b_16h",
+    "mobilenet_v2_quantized_qat",
+    "phi_1_5",
+    "resnet50_quantized_qat",
+    "DALLE2_pytorch",
+    "moco",
+    "timm_efficientdet",
+    "pytorch_CycleGAN_and_pix2pix",
+    "vision_maskrcnn",
+    "dlrm",
+    "resnet50",
+    *DETECTRON2_MODELS,
 }
 
 # Skip the experiment of a model if any of the experiment configs in the list is fully matched