format code

pyproject.toml

@@ -16,6 +16,7 @@ include = '\.pyi?$'
 py_version = 38
 line_length = 120
 profile = 'black'
+known_third_party = ["nle"]
 
 [tool.pytest.ini_options]
 addopts = "-s"

sample_factory/algo/utils/model_sharing.py

@@ -1,6 +1,7 @@
 """
 Utilities for sharing model parameters between components.
 """
+
 import sys
 from typing import Optional
 

sample_factory/algo/utils/running_mean_std.py

@@ -3,6 +3,7 @@
 All credit goes to https://github.com/Denys88/rl_games (only slightly changed here, mostly with in-place operations)
 Thanks a lot, great module!
 """
+
 from typing import Dict, Final, List, Optional, Union
 
 import gymnasium as gym

sample_factory/envs/env_wrappers.py

@@ -2,6 +2,7 @@
 Gym env wrappers that make the environment suitable for the RL algorithms.
 
 """
+
 import json
 import os
 from os.path import join

sample_factory/launcher/run_slurm.py

@@ -4,7 +4,6 @@
 
 """
 
-
 import os
 import time
 from os.path import join

sample_factory/pbt/population_based_training.py

@@ -1,6 +1,5 @@
 """Population-Based Training implementation, inspired by https://arxiv.org/abs/1807.01281."""
 
-
 import copy
 import json
 import math

sample_factory/utils/normalize.py

@@ -10,6 +10,7 @@
 If no data normalization is needed we just keep the original data.
 Otherwise, we create a copy of data and do all of the operations operations in-place.
 """
+
 from typing import Dict
 
 import torch

sf_examples/brax/brax_render.py

@@ -6,7 +6,6 @@
 render_mode=human (although still slow and in low resolution).
 """
 
-
 from typing import Dict, List, Tuple
 
 import brax

sf_examples/brax/train_brax.py

@@ -1,6 +1,7 @@
 """
 Brax env integration.
 """
+
 import sys
 from typing import Dict, List, Optional, Tuple, Union
 

sf_examples/nethack/models/chaotic_dwarf.py

@@ -23,6 +23,7 @@
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.
 """
+
 import torch
 from nle import nethack
 from torch import nn

sf_examples/nethack/nethack_env.py

@@ -10,6 +10,7 @@
     NetHackStaircase,
     NetHackStaircasePet,
 )
+
 from sample_factory.algo.utils.gymnasium_utils import patch_non_gymnasium_env
 from sf_examples.nethack.utils.wrappers import (
     BlstatsInfoWrapper,

sf_examples/nethack/utils/wrappers/gym_compatibility.py

@@ -1,4 +1,5 @@
 """Compatibility wrappers for OpenAI gym V22 and V26."""
+
 # pyright: reportGeneralTypeIssues=false, reportPrivateImportUsage=false
 from __future__ import annotations
 

sf_examples/nethack/utils/wrappers/timelimit.py

@@ -1,5 +1,4 @@
 import gym
-
 from nle.env.base import NLE
 
 

sf_examples/train_custom_env_custom_model.py

@@ -6,6 +6,7 @@
 python -m sf_examples.enjoy_custom_env_custom_model --algo=APPO --env=my_custom_env_v1 --experiment=example
 
 """
+
 from __future__ import annotations
 
 import sys

sf_examples/train_custom_multi_env.py

@@ -6,6 +6,7 @@
 python -m sf_examples.enjoy_custom_multi_env --algo=APPO --env=my_custom_multi_env_v1 --experiment=example_multi
 
 """
+
 from __future__ import annotations
 
 import random

sf_examples/vizdoom/train_custom_vizdoom_env.py

@@ -8,6 +8,7 @@
 python -m sf_examples.vizdoom.enjoy_custom_vizdoom_env --algo=APPO --env=doom_my_custom_env --experiment=doom_my_custom_env_example
 
 """
+
 import argparse
 import functools
 import os