From 6b3ee104fb0b9edf856c92ba69baa4bb25dd5969 Mon Sep 17 00:00:00 2001
From: Ikechukwu Uchendu <iuchendu@g.harvard.edu>
Date: Tue, 3 Sep 2024 11:56:02 -0400
Subject: [PATCH] Fixed formatting

---
 a2perf/launch/entrypoint.py | 14 ++++++++------
 setup.py                    |  7 -------
 2 files changed, 8 insertions(+), 13 deletions(-)

diff --git a/a2perf/launch/entrypoint.py b/a2perf/launch/entrypoint.py
index d07113b..b8b0eb9 100644
--- a/a2perf/launch/entrypoint.py
+++ b/a2perf/launch/entrypoint.py
@@ -1,9 +1,7 @@
 import os
 import subprocess
 
-from absl import app
-from absl import flags
-from absl import logging
+from absl import app, flags, logging
 
 _ROOT_DIR = flags.DEFINE_string(
     "root-dir", None, "Root directory to save metrics and logs from training"
@@ -22,18 +20,22 @@
 _PARTICIPANT_ARGS = flags.DEFINE_string(
     "participant-args",
     None,
-    "Additional keyword arguments to pass to the participant's train and inference functions.",
+    "Additional keyword arguments to pass to the"
+    " participant's train and inference functions.",
 )
 
 
 def _usage():
     return (
-        "Usage: a2perf <participant_module_path> --root-dir=<root_dir> --submission-gin-config-path=<path> [--participant-args=<k1=v1,k2=v2,...>]\n"
+        "Usage: a2perf <participant_module_path> "
+        "--root-dir=<root_dir> "
+        "--submission-gin-config-path=<path> "
+        "[--participant-args=<k1=v1,k2=v2,...>]\n"
         "\n"
         "Options:\n"
         "\t--root-dir\tRoot directory for the experiment\n"
         "\t--submission-gin-config-path\tPath to the gin configuration file\n"
-        "\t--participant-args\tAdditional arguments for the participant's train function"
+        "\t--participant-args\tAdditional arguments for the participant's train function"  # noqa: E501
     )
 
 
diff --git a/setup.py b/setup.py
index 61fcdfb..3610ea0 100644
--- a/setup.py
+++ b/setup.py
@@ -1,14 +1,7 @@
 """Sets up the project."""
 
-import importlib.util
 import logging
-import os
 import pathlib
-import shutil
-import site
-import sys
-import urllib.request
-from distutils.cmd import Command
 
 from setuptools import setup