pyrefly

Signed-off-by: Yuki Huang <[email protected]>

Author: yuki-97

pyrefly.toml

@@ -34,10 +34,12 @@ project-includes = [
     "docs/helpers.py",
     "examples/converters/convert_dcp_to_hf.py",
     "examples/converters/convert_megatron_to_hf.py",
+    "examples/custom_parallel/custom_parallel.py",
+    "examples/custom_parallel/llama_nemotron_super_49b_custom_plan.py",
     "nemo_rl/algorithms/__init__.py",
     "nemo_rl/algorithms/interfaces.py",
-    "nemo_rl/algorithms/utils.py",
     "nemo_rl/algorithms/reward_functions.py",
+    "nemo_rl/algorithms/utils.py",
     "nemo_rl/data/__init__.py",
     "nemo_rl/data/chat_templates.py",
     "nemo_rl/data/collate_fn.py",
@@ -65,6 +67,7 @@ project-includes = [
     "nemo_rl/data/datasets/response_datasets/refcoco.py",
     "nemo_rl/data/datasets/response_datasets/response_dataset.py",
     "nemo_rl/data/datasets/response_datasets/squad.py",
+    "nemo_rl/data/datasets/response_datasets/tulu3.py",
     "nemo_rl/data/datasets/utils.py",
     "nemo_rl/data/interfaces.py",
     "nemo_rl/data/multimodal_utils.py",
@@ -78,6 +81,7 @@ project-includes = [
     "nemo_rl/distributed/worker_group_utils.py",
     "nemo_rl/environments/__init__.py",
     "nemo_rl/environments/games/sliding_puzzle.py",
+    "nemo_rl/environments/helpsteer3_environment.py",
     "nemo_rl/environments/interfaces.py",
     "nemo_rl/environments/math_environment.py",
     "nemo_rl/environments/metrics.py",
@@ -104,10 +108,10 @@ project-includes = [
     "nemo_rl/models/policy/interfaces.py",
     "nemo_rl/models/policy/utils.py",
     "nemo_rl/utils/__init__.py",
+    "nemo_rl/utils/automodel_checkpoint.py",
     "nemo_rl/utils/checkpoint.py",
     "nemo_rl/utils/config.py",
     "nemo_rl/utils/native_checkpoint.py",
-    "nemo_rl/utils/automodel_checkpoint.py",
     "nemo_rl/utils/nsys.py",
     "nemo_rl/utils/nvml.py",
     "nemo_rl/utils/packed_tensor.py",