fix ruff formatting

Author: arhanjain

src/openpi/training/misc/polaris_config.py

@@ -31,7 +31,8 @@ def get_polaris_configs():
             model=pi0_config.Pi0Config(action_horizon=15, pi05=True),
             data=RLDSDroidDataConfig(
                 assets=AssetsConfig(
-                    assets_dir="gs://openpi-assets/checkpoints/polaris/pi05_droid_jointpos_polaris/assets", asset_id="droid"
+                    assets_dir="gs://openpi-assets/checkpoints/polaris/pi05_droid_jointpos_polaris/assets",
+                    asset_id="droid",
                 ),
                 datasets=(
                     droid_rlds_dataset.RLDSDataset(
@@ -75,7 +76,8 @@ def get_polaris_configs():
             ),
             data=RLDSDroidDataConfig(
                 assets=AssetsConfig(
-                    assets_dir="gs://openpi-assets/checkpoints/polaris/pi0_fast_droid_jointpos_polaris/assets", asset_id="droid"
+                    assets_dir="gs://openpi-assets/checkpoints/polaris/pi0_fast_droid_jointpos_polaris/assets",
+                    asset_id="droid",
                 ),
                 datasets=(
                     droid_rlds_dataset.RLDSDataset(
@@ -119,7 +121,8 @@ def get_polaris_configs():
             ),
             data=RLDSDroidDataConfig(
                 assets=AssetsConfig(
-                    assets_dir="gs://openpi-assets/checkpoints/polaris/pi0_droid_jointpos_polaris/assets", asset_id="droid"
+                    assets_dir="gs://openpi-assets/checkpoints/polaris/pi0_droid_jointpos_polaris/assets",
+                    asset_id="droid",
                 ),
                 datasets=(
                     droid_rlds_dataset.RLDSDataset(
@@ -138,7 +141,9 @@ def get_polaris_configs():
                 rlds_data_dir="<path_to_droid_rlds_dataset>",
                 action_space=droid_rlds_dataset.DroidActionSpace.JOINT_POSITION,
             ),
-            weight_loader=weight_loaders.CheckpointWeightLoader("gs://openpi-assets/checkpoints/polaris/pi0_droid_jointpos_polaris/params"),
+            weight_loader=weight_loaders.CheckpointWeightLoader(
+                "gs://openpi-assets/checkpoints/polaris/pi0_droid_jointpos_polaris/params"
+            ),
             lr_schedule=_optimizer.CosineDecaySchedule(
                 warmup_steps=1_000,
                 peak_lr=5e-5,
@@ -161,7 +166,8 @@ def get_polaris_configs():
             ),
             data=RLDSDroidDataConfig(
                 assets=AssetsConfig(
-                    assets_dir="gs://openpi-assets/checkpoints/polaris/pi0_droid_jointpos_100k_polaris/assets", asset_id="droid"
+                    assets_dir="gs://openpi-assets/checkpoints/polaris/pi0_droid_jointpos_100k_polaris/assets",
+                    asset_id="droid",
                 ),
                 datasets=(
                     droid_rlds_dataset.RLDSDataset(