[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

lerobot/common/datasets/v21/convert_dataset_v20_to_v21.py

@@ -38,7 +38,10 @@ from huggingface_hub import HfApi
 
 from lerobot.common.datasets.lerobot_dataset import CODEBASE_VERSION, LeRobotDataset
 from lerobot.common.datasets.utils import EPISODES_STATS_PATH, STATS_PATH, load_stats, write_info
-from lerobot.common.datasets.v21.convert_stats import check_aggregate_stats, convert_stats, convert_stats_parallel
+from lerobot.common.datasets.v21.convert_stats import (
+    check_aggregate_stats,
+    convert_stats_parallel,
+)
 
 V20 = "v2.0"
 V21 = "v2.1"

lerobot/common/datasets/v21/convert_stats.py

@@ -13,11 +13,10 @@
 # limitations under the License.
 
 from concurrent.futures import ThreadPoolExecutor, as_completed
+from multiprocessing import cpu_count
 
 import numpy as np
 from tqdm import tqdm
-from multiprocessing import cpu_count
-from concurrent.futures import ProcessPoolExecutor, as_completed
 
 from lerobot.common.datasets.compute_stats import aggregate_stats, get_feature_stats, sample_indices
 from lerobot.common.datasets.lerobot_dataset import LeRobotDataset
@@ -91,9 +90,7 @@ def convert_stats_parallel(dataset: LeRobotDataset, num_workers: int = 0):
     if num_workers > 0:
         with ThreadPoolExecutor(max_workers=max_workers) as executor:
             for ep_idx in range(total_episodes):
-                futures.append(
-                    executor.submit(convert_episode_stats, dataset, ep_idx, True)
-                )
+                futures.append(executor.submit(convert_episode_stats, dataset, ep_idx, True))
             for future in tqdm(as_completed(futures), total=total_episodes, desc="Converting episodes stats"):
                 ep_stats, ep_data = future.result()
                 dataset.meta.episodes_stats[ep_idx] = ep_data