Convert directory fbcode/aibench to use the Ruff Formatter

Summary:
Converts the directory specified to use the Ruff formatter in pyfmt

ruff_dog

If this diff causes merge conflicts when rebasing, please run
`hg status -n -0 --change . -I '**/*.{py,pyi}' | xargs -0 arc pyfmt`
on your diff, and amend any changes before rebasing onto latest.
That should help reduce or eliminate any merge conflicts.

allow-large-files

Reviewed By: amyreese

Differential Revision: D66250245

fbshipit-source-id: 91d6f48b32e47a33841aaefe1e2b43b7705c1267

Author: Thomas Polasek

ailab/benchmark/db_controller.py

@@ -138,7 +138,9 @@ def get_payload(req):
                 "status": (
                     "OCCUPIED"
                     if d[2] == "0"
-                    else "AVAILABLE" if d[2] == "1" else "OFFLINE"
+                    else "AVAILABLE"
+                    if d[2] == "1"
+                    else "OFFLINE"
                 ),
                 "claimer": claimer,
                 "job_queue": job_queue,

ailab/file_storage/views.py

@@ -9,7 +9,6 @@
 
 @csrf_exempt
 def upload(request):
-
     # Handle file upload
     if request.method == "POST" and "file" in request.FILES:
         file = request.FILES["file"]

benchmarking/benchmarks/benchmarks.py

@@ -30,7 +30,6 @@
 
 class BenchmarkCollector:
     def __init__(self, framework, model_cache, **kwargs):
-
         self.args = kwargs.get("args", None)
         if not os.path.isdir(model_cache):
             os.makedirs(model_cache)
@@ -104,7 +103,6 @@ def _collectOneBenchmark(self, source, meta, benchmarks, info, user_identifier):
     # Update all files in the benchmark to absolute path
     # download the files if needed
     def _updateFiles(self, one_benchmark, filename, user_identifier):
-
         model = one_benchmark["model"]
         model_dir = os.path.join(self.model_cache, model["format"], model["name"])
         if not os.path.isdir(model_dir):

benchmarking/data_converters/json_converter/json_converter.py

@@ -79,8 +79,7 @@ def convert(self, data):
                         new_string = d["info_string"]
                         if old_string != new_string:
                             getLogger().warning(
-                                "info_string values "
-                                "for {} ".format(key)
+                                "info_string values " "for {} ".format(key)
                                 + "do not match.\n"
                                 + "Current info_string: "
                                 + "{}\n ".format(old_string)

benchmarking/download_benchmarks/download_benchmarks.py

@@ -52,18 +52,18 @@ def _processOneBenchmark(self, benchmark):
             if "files" in one_benchmark["model"]:
                 for field in one_benchmark["model"]["files"]:
                     value = one_benchmark["model"]["files"][field]
-                    assert (
-                        "location" in value
-                    ), "location field is missing in benchmark " "{}".format(filename)
+                    assert "location" in value, (
+                        "location field is missing in benchmark " "{}".format(filename)
+                    )
                     location = value["location"]
                     md5 = value.get("md5")
                     path = self.downloadFile(location, md5)
                     locations.append(path)
             if "libraries" in one_benchmark["model"]:
                 for value in one_benchmark["model"]["libraries"]:
-                    assert (
-                        "location" in value
-                    ), "location field is missing in benchmark " "{}".format(filename)
+                    assert "location" in value, (
+                        "location field is missing in benchmark " "{}".format(filename)
+                    )
                     location = value["location"]
                     md5 = value.get("md5")
                     path = self.downloadFile(location, md5)

benchmarking/frameworks/caffe2/caffe2.py

@@ -55,23 +55,20 @@ def verifyBenchmarkFile(self, benchmark, filename, is_post):
 
             for f in model["files"]:
                 field = model["files"][f]
-                assert (
-                    "filename" in field
-                ), "Filename is missing in file" + " {} of benchmark {}".format(
-                    f, filename
+                assert "filename" in field, (
+                    "Filename is missing in file"
+                    + " {} of benchmark {}".format(f, filename)
                 )
-                assert (
-                    "location" in field
-                ), "Location is missing in file" + " {} of benchmark {}".format(
-                    f, filename
+                assert "location" in field, (
+                    "Location is missing in file"
+                    + " {} of benchmark {}".format(f, filename)
                 )
                 if "md5" not in field:
                     assert not field["location"].startswith("//") or field[
                         "location"
-                    ].startswith(
-                        "//fbpkg"
-                    ), "MD5 is missing in file" + " {} of benchmark {}".format(
-                        f, filename
+                    ].startswith("//fbpkg"), (
+                        "MD5 is missing in file"
+                        + " {} of benchmark {}".format(f, filename)
                     )
 
         # tests is mandatory
@@ -90,9 +87,9 @@ def verifyBenchmarkFile(self, benchmark, filename, is_post):
         is_generic_test = tests[0]["metric"] == "generic"
 
         for test in tests:
-            assert (
-                "metric" in test
-            ), "Metric field is missing in " + "benchmark {}".format(filename)
+            assert "metric" in test, (
+                "Metric field is missing in " + "benchmark {}".format(filename)
+            )
 
             # no check is needed if the metric is generic
             if is_generic_test:
@@ -104,16 +101,16 @@ def verifyBenchmarkFile(self, benchmark, filename, is_post):
             if "warmup" not in test:
                 test["warmup"] = -1
 
-            assert (
-                "identifier" in test
-            ), "Identifier field is missing in " + "benchmark {}".format(filename)
+            assert "identifier" in test, (
+                "Identifier field is missing in " + "benchmark {}".format(filename)
+            )
 
             if "commands" in test or "command" in test or "arguments" in test:
                 continue
             # for backward compatibility purpose
-            assert (
-                "inputs" in test
-            ), "Inputs field is missing in " + "benchmark {}".format(filename)
+            assert "inputs" in test, (
+                "Inputs field is missing in " + "benchmark {}".format(filename)
+            )
 
             num = -1
             for ip_name in test["inputs"]:

benchmarking/frameworks/framework_base.py

@@ -577,7 +577,9 @@ def _runCommands(
                 else (
                     "simpleperf"
                     if ["cpu"] == types
-                    else "perfetto" if PerfettoAnySupported(types) else "<unspecified>"
+                    else "perfetto"
+                    if PerfettoAnySupported(types)
+                    else "<unspecified>"
                 )
             )
             profiler = profiling_args.setdefault("profiler", default_profiler)

benchmarking/harness.py

@@ -253,10 +253,12 @@ def runBenchmark(self, info, platform, benchmarks):
             # check the framework matches
             if "model" in benchmark and "framework" in benchmark["model"]:
                 assert benchmark["model"]["framework"] == self.args.framework, (
-                    "Framework specified in the json file "
-                    "{} ".format(benchmark["model"]["framework"])
-                    + "does not match the command line argument "
-                    "{}".format(self.args.framework)
+                    "Framework specified in the json file " "{} ".format(
+                        benchmark["model"]["framework"]
+                    )
+                    + "does not match the command line argument " "{}".format(
+                        self.args.framework
+                    )
                 )
             if self.args.debug:
                 for test in benchmark["tests"]:

benchmarking/platforms/device_manager.py

@@ -423,7 +423,7 @@ def run(self):
             getLogger().info(
                 f"\nBattery status: {battery_state['status']}"
                 + f"\nBattery charge level: {battery_state['charge_level']}%"
-                + f"\nBattery temperature: {battery_state['temperature']}\xB0C"
+                + f"\nBattery temperature: {battery_state['temperature']}\xb0C"
             )
             if battery_state["disconnected"]:
                 getLogger().warning(
@@ -456,9 +456,10 @@ def run(self):
             battery_state = getBatteryState(
                 self.device["hash"], self.args.platform, self.args.android_dir
             )
-            if (battery_state["supported"]) and battery_state[
-                "charge_level"
-            ] < charge_threshold:
+            if (
+                (battery_state["supported"])
+                and battery_state["charge_level"] < charge_threshold
+            ):
                 getLogger().info(
                     f"Battery charge of {battery_state['charge_level']}% is below threshold of {charge_threshold}%; Sleep another {self.cooldown} seconds."
                 )

benchmarking/profilers/perfetto/perfetto.py

@@ -62,7 +62,6 @@ class BatteryState(Enum):
 
 
 class Perfetto(ProfilerBase):
-
     CONFIG_FILE = "perfetto.conf"
     DEVICE_DIRECTORY = "/data/local/tmp/perf"
     DEVICE_TRACE_DIRECTORY = "/data/misc/perfetto-traces"