Sort __all__

Stable-Baselines-Team · Dec 2, 2024 · 53baa5d · 53baa5d
1 parent 774ac18
commit 53baa5d
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 8 deletions.
diff --git a/sb3_contrib/__init__.py b/sb3_contrib/__init__.py
@@ -15,10 +15,10 @@
 
 __all__ = [
     "ARS",
-    "CrossQ",
-    "MaskablePPO",
-    "RecurrentPPO",
     "QRDQN",
     "TQC",
     "TRPO",
+    "CrossQ",
+    "MaskablePPO",
+    "RecurrentPPO",
 ]
diff --git a/sb3_contrib/common/torch_layers.py b/sb3_contrib/common/torch_layers.py
@@ -1,6 +1,6 @@
 import torch
 
-__all__ = ["BatchRenorm1d", "BatchRenorm"]
+__all__ = ["BatchRenorm", "BatchRenorm1d"]
 
 
 class BatchRenorm(torch.nn.Module):

diff --git a/sb3_contrib/ppo_mask/__init__.py b/sb3_contrib/ppo_mask/__init__.py
@@ -1,4 +1,4 @@
 from sb3_contrib.ppo_mask.policies import CnnPolicy, MlpPolicy, MultiInputPolicy
 from sb3_contrib.ppo_mask.ppo_mask import MaskablePPO
 
-__all__ = ["CnnPolicy", "MlpPolicy", "MultiInputPolicy", "MaskablePPO"]
+__all__ = ["CnnPolicy", "MaskablePPO", "MlpPolicy", "MultiInputPolicy"]
diff --git a/sb3_contrib/qrdqn/__init__.py b/sb3_contrib/qrdqn/__init__.py
@@ -1,4 +1,4 @@
 from sb3_contrib.qrdqn.policies import CnnPolicy, MlpPolicy, MultiInputPolicy
 from sb3_contrib.qrdqn.qrdqn import QRDQN
 
-__all__ = ["CnnPolicy", "MlpPolicy", "MultiInputPolicy", "QRDQN"]
+__all__ = ["QRDQN", "CnnPolicy", "MlpPolicy", "MultiInputPolicy"]
diff --git a/sb3_contrib/tqc/__init__.py b/sb3_contrib/tqc/__init__.py
@@ -1,4 +1,4 @@
 from sb3_contrib.tqc.policies import CnnPolicy, MlpPolicy, MultiInputPolicy
 from sb3_contrib.tqc.tqc import TQC
 
-__all__ = ["CnnPolicy", "MlpPolicy", "MultiInputPolicy", "TQC"]
+__all__ = ["TQC", "CnnPolicy", "MlpPolicy", "MultiInputPolicy"]
diff --git a/sb3_contrib/trpo/__init__.py b/sb3_contrib/trpo/__init__.py
@@ -1,4 +1,4 @@
 from sb3_contrib.trpo.policies import CnnPolicy, MlpPolicy, MultiInputPolicy
 from sb3_contrib.trpo.trpo import TRPO
 
-__all__ = ["CnnPolicy", "MlpPolicy", "MultiInputPolicy", "TRPO"]
+__all__ = ["TRPO", "CnnPolicy", "MlpPolicy", "MultiInputPolicy"]