new black codestyle

Author: ymetz

rlhfblender/data_collection/feedback_translator.py

@@ -2,7 +2,6 @@
 This module translates incoming feedback of different types into a common format.
 """
 
-
 from rlhfblender.data_models.feedback_models import (
     AbsoluteFeedback,
     Actuality,

rlhfblender/data_collection/sampler.py

@@ -1,6 +1,7 @@
 """
 A sampler for episodes for HITL experiments
 """
+
 import os
 from enum import Enum
 from typing import List, Optional

rlhfblender/generate_data.py

@@ -2,6 +2,7 @@
 Pre-generated data for the RLHFBlender project (e.g. record videos, reward logs, etc.)
 These can the be loaded in the user interface for studies
 """
+
 import argparse
 import asyncio
 import sys

rlhfblender/logger/logger.py

@@ -1,6 +1,7 @@
 """
 Logging component for feedback logging
 """
+
 from abc import abstractmethod
 from datetime import datetime
 from typing import List

rlhfblender/startup_script.py

@@ -3,6 +3,7 @@
 Expects pre-trained models in the experimentation directory. First runs benchmarks with the provided
 models, then creates video/thumbnail/reward data etc.
 """
+
 import asyncio
 import os
 

rlhfblender/utils/data_generation.py

@@ -3,6 +3,7 @@
 Expects pre-trained models in the experimentation directory. First runs benchmarks with the provided
 models, then creates video/thumbnail/reward data etc.
 """
+
 import os
 import time
 from types import SimpleNamespace as sn
@@ -76,9 +77,9 @@ async def run_benchmark(request: List[BenchmarkRequestModel]) -> list[Experiment
                 # We lazily register the environment if it is not registered yet, this is only done once
                 database_env = initial_registration(
                     benchmark_run.env_id,
-                    additional_gym_packages=benchmark_run.additional_packages
-                    if "additional_packages" in benchmark_run
-                    else [],
+                    additional_gym_packages=(
+                        benchmark_run.additional_packages if "additional_packages" in benchmark_run else []
+                    ),
                 )
                 await db_handler.add_entry(database, Environment, database_env.model_dump())
 

tests/test_data_endpoints.py

@@ -3,6 +3,7 @@
     Test the DATA API endpoints
 
 """
+
 import os
 
 from fastapi.testclient import TestClient