Fixes order of imports for added standard modules (isaac-sim#4328)

# Description

We now consider: "warp" and "pytest" as a standard library.

## Type of change

- New feature (non-breaking change which adds functionality)

## Checklist

- [x] I have read and understood the [contribution
guidelines](https://isaac-sim.github.io/IsaacLab/main/source/refs/contributing.html)
- [x] I have run the [`pre-commit` checks](https://pre-commit.com/) with
`./isaaclab.sh --format`
- [ ] I have made corresponding changes to the documentation
- [x] My changes generate no new warnings
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have updated the changelog and the corresponding version in the
extension's `config/extension.toml` file
- [x] I have added my name to the `CONTRIBUTORS.md` or my name already
exists there

Author: Mayankm96

.pre-commit-config.yaml

@@ -8,7 +8,7 @@ repos:
     rev: 24.3.0
     hooks:
       - id: black
-        args: ["--line-length", "120", "--unstable"]
+        args: ["--unstable"]
   - repo: https://github.com/pycqa/flake8
     rev: 7.0.0
     hooks:
@@ -36,7 +36,6 @@ repos:
     hooks:
       - id: isort
         name: isort (python)
-        args: ["--profile", "black", "--filter-files"]
   - repo: https://github.com/asottile/pyupgrade
     rev: v3.15.1
     hooks:

docker/test/test_docker.py

@@ -4,11 +4,10 @@
 # SPDX-License-Identifier: BSD-3-Clause
 
 import os
+import pytest
 import subprocess
 from pathlib import Path
 
-import pytest
-
 
 def start_stop_docker(profile, suffix):
     """Test starting and stopping docker profile with suffix."""

pyproject.toml

@@ -1,5 +1,16 @@
+# Copyright (c) 2022-2026, The Isaac Lab Project Developers (https://github.com/isaac-sim/IsaacLab/blob/main/CONTRIBUTORS.md).
+# All rights reserved.
+#
+# SPDX-License-Identifier: BSD-3-Clause
+
+[tool.black]
+line-length = 120
+target-version = ["py311"]
+preview = true
+
 [tool.isort]
 
+profile = "black"
 py_version = 310
 line_length = 120
 group_by_package = true
@@ -11,6 +22,7 @@ skip_glob = ["docs/*", "logs/*", "_isaac_sim/*", ".vscode/*"]
 sections = [
     "FUTURE",
     "STDLIB",
+    "OMNIVERSE_EXTENSIONS",
     "THIRDPARTY",
     "ASSETS_FIRSTPARTY",
     "FIRSTPARTY",
@@ -22,44 +34,63 @@ sections = [
 # Extra standard libraries considered as part of python (permissive licenses
 extra_standard_library = [
     "numpy",
-    "h5py",
-    "open3d",
     "torch",
+    "einops",
     "tensordict",
+    "warp",
+    "scipy",
+    "open3d",
+    "cv2",
+    "PIL",
+    "torchvision",
+    "transformers",
+    "h5py",
+    "yaml",
+    "toml",
     "bpy",
+    "trimesh",
     "matplotlib",
     "gymnasium",
     "gym",
-    "scipy",
     "hid",
-    "yaml",
     "prettytable",
-    "toml",
-    "trimesh",
     "tqdm",
-    "torchvision",
-    "transformers",
-    "einops", # Needed for transformers, doesn't always auto-install
+    "flatdict",
     "packaging",
+    "pytest",
+    "pytest-mock",
+    "flaky",
 ]
-# Imports from Isaac Sim and Omniverse
+
 known_third_party = [
-    "isaacsim.core.api",
-    "isaacsim.replicator.common",
-    "omni.replicator.core",
-    "pxr",
-    "omni.kit.*",
-    "warp",
+    "junitparser",
+    "pinocchio",
+    "pink",
+    "rsl_rl",
+    "rl_games",
+    "ray",
+    "stable_baselines3",
+    "skrl",
+    "wandb",
+    "tensorboard",
+    "hydra",
+    "omegaconf",
+]
+
+# Imports from Isaac Sim and Omniverse
+known_omniverse_extensions = [
     "carb",
+    "omni*",
+    "isaacsim*",
+    "pxr",
+    "usdrt",
     "Semantics",
 ]
+
 # Imports from this repository
 known_first_party = "isaaclab"
 known_assets_firstparty = "isaaclab_assets"
-known_extra_firstparty = [
-    "isaaclab_rl",
-    "isaaclab_mimic",
-]
+known_extra_firstparty = ["isaaclab_rl", "isaaclab_mimic"]
 known_task_firstparty = "isaaclab_tasks"
 # Imports from the local folder
 known_local_folder = "config"

scripts/benchmarks/utils.py

@@ -9,6 +9,7 @@
 
 from isaacsim.benchmark.services import BaseIsaacBenchmark
 from isaacsim.benchmark.services.metrics.measurements import DictMeasurement, ListMeasurement, SingleMeasurement
+
 from tensorboard.backend.event_processing import event_accumulator
 
 

scripts/demos/h1_locomotion.py

@@ -48,6 +48,7 @@
 from omni.kit.viewport.utility import get_viewport_from_window_name
 from omni.kit.viewport.utility.camera_state import ViewportCameraState
 from pxr import Gf, Sdf
+
 from rsl_rl.runners import OnPolicyRunner
 
 from isaaclab.envs import ManagerBasedRLEnv

scripts/environments/state_machine/lift_cube_sm.py

@@ -40,9 +40,8 @@
 
 import gymnasium as gym
 import torch
-from collections.abc import Sequence
-
 import warp as wp
+from collections.abc import Sequence
 
 from isaaclab.assets.rigid_object.rigid_object_data import RigidObjectData
 

scripts/environments/state_machine/lift_teddy_bear.py

@@ -42,9 +42,8 @@
 
 import gymnasium as gym
 import torch
-from collections.abc import Sequence
-
 import warp as wp
+from collections.abc import Sequence
 
 from isaaclab.assets.rigid_object.rigid_object_data import RigidObjectData
 

scripts/environments/state_machine/open_cabinet_sm.py

@@ -40,9 +40,8 @@
 
 import gymnasium as gym
 import torch
-from collections.abc import Sequence
-
 import warp as wp
+from collections.abc import Sequence
 
 from isaaclab.sensors import FrameTransformer
 

scripts/tools/hdf5_to_mp4.py

@@ -21,16 +21,12 @@
     --framerate          Frames per second for the output video. (default: 30)
 """
 
-# Standard library imports
 import argparse
+import cv2
 import h5py
 import numpy as np
-
-# Third-party imports
 import os
 
-import cv2
-
 # Constants
 DEFAULT_VIDEO_HEIGHT = 704
 DEFAULT_VIDEO_WIDTH = 1280

scripts/tools/mp4_to_hdf5.py

@@ -17,17 +17,13 @@
     --videos_dir         Directory containing the visually augmented MP4 videos.
 """
 
-# Standard library imports
 import argparse
+import cv2
 import glob
 import h5py
 import numpy as np
-
-# Third-party imports
 import os
 
-import cv2
-
 
 def parse_args():
     """Parse command line arguments."""