style: apply pre-commit auto-fixes (ruff)

Drops stray blank lines after import blocks (E303) and reorders local
imports per isort across data_plane adapters/policy plus the
data_plane test suite.

Pure auto-fix output from pre-commit run --all-files; no semantic
changes.

Signed-off-by: Zhiyu Li <zhiyul@NVIDIA.com>

Author: ZhiyuLi-Nvidia

nemo_rl/data_plane/adapters/transfer_queue.py

@@ -40,7 +40,6 @@
     KVBatchMeta,
 )
 
-
 # ──────────────────────────────────────────────────────────────────────────
 # Lazy import of transfer_queue — keeps NeMo-RL importable without TQ
 # installed; failure is deferred to construction time.

nemo_rl/models/policy/tq_policy.py

@@ -53,7 +53,6 @@
 from nemo_rl.utils.flops_tracker import get_theoretical_tflops
 from nemo_rl.utils.timer import Timer
 
-
 # ──────────────────────────────────────────────────────────────────────────
 # Per-stage aggregators that assemble per-rank worker results into the
 # shape each Policy method returns. Used by the TQ-mediated overrides

tests/data_plane/functional/test_seqpack_equivalence.py

@@ -48,7 +48,6 @@
 from nemo_rl.data_plane import build_data_plane_client, materialize
 from nemo_rl.distributed.batched_data_dict import BatchedDataDict
 
-
 # Mirror of the seed-field set in nemo_rl/algorithms/grpo_sync.py.
 _DP_SEED_FIELDS = (
     "input_ids",

tests/data_plane/functional/test_tq_multinode.py

@@ -24,7 +24,6 @@
 
 from __future__ import annotations
 
-
 import pytest
 import torch
 from tensordict import TensorDict

tests/data_plane/unit/test_codec_mooncake.py

@@ -25,7 +25,6 @@
 import pytest
 import torch
 
-
 # ── Module-level state restoration fixture ───────────────────────────────────
 
 
@@ -55,9 +54,10 @@ def test_promote_1d_unsqueezes_on_write(codec_flags) -> None:
     pre-unsqueeze at the wire layer so per-row shape matches the metadata shape.
     """
     # Import the adapter's _to_wire directly so this test stays unit-level.
-    from nemo_rl.data_plane.adapters.transfer_queue import _to_wire
     from tensordict import TensorDict
 
+    from nemo_rl.data_plane.adapters.transfer_queue import _to_wire
+
     codec_flags.set_kv_promote_1d(True)
 
     n = 8
@@ -106,9 +106,10 @@ def test_promote_1d_squeezes_on_read_roundtrip(codec_flags) -> None:
 def test_promote_1d_off_leaves_shape_unchanged(codec_flags) -> None:
     """When _KV_PROMOTE_1D is False (the default), 1D tensors pass through
     the wire layer without modification."""
-    from nemo_rl.data_plane.adapters.transfer_queue import _to_wire
     from tensordict import TensorDict
 
+    from nemo_rl.data_plane.adapters.transfer_queue import _to_wire
+
     codec_flags.set_kv_promote_1d(False)
 
     n = 5

tests/data_plane/unit/test_correctness.py

@@ -25,13 +25,12 @@
 import torch
 from tensordict import TensorDict
 
-from nemo_rl.experience.sync_rollout_actor import kv_first_write
 from nemo_rl.data_plane.adapters.noop import NoOpDataPlaneClient
 from nemo_rl.data_plane.driver_io import read_columns, write_columns
 from nemo_rl.data_plane.interfaces import KVBatchMeta
 from nemo_rl.data_plane.preshard import DP_SEED_FIELDS, shard_meta_for_dp
 from nemo_rl.distributed.batched_data_dict import BatchedDataDict
-
+from nemo_rl.experience.sync_rollout_actor import kv_first_write
 
 # ── helpers ────────────────────────────────────────────────────────────
 

tests/data_plane/unit/test_interface_contract.py

@@ -20,7 +20,6 @@
 
 from __future__ import annotations
 
-
 import pytest
 import torch
 from tensordict import TensorDict

tests/data_plane/unit/test_leader_broadcast.py

@@ -25,8 +25,8 @@
 import torch.distributed as dist
 import torch.multiprocessing as mp
 
-from nemo_rl.distributed.batched_data_dict import BatchedDataDict
 from nemo_rl.data_plane.worker_mixin import _broadcast_batched_data_dict
+from nemo_rl.distributed.batched_data_dict import BatchedDataDict
 
 
 def _worker(rank: int, world_size: int, tmp_init_file: str, q):

tests/data_plane/unit/test_local_node_ip.py

@@ -28,7 +28,6 @@
 
 import pytest
 
-
 # ── helpers ──────────────────────────────────────────────────────────────────
 
 
@@ -126,6 +125,7 @@ def test_mc_tcp_bind_address_overwrites_existing(monkeypatch) -> None:
     must be overwritten with the local IP.
     """
     import socket
+
     from nemo_rl.data_plane.adapters.transfer_queue import _get_local_node_ip
 
     local_ip = "10.65.4.100"