Apply format to runner.py in torch

Toni-SM · Nov 3, 2024 · b65872c · b65872c
1 parent 0dc36ea
commit b65872c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/skrl/utils/runner/torch/runner.py b/skrl/utils/runner/torch/runner.py
@@ -14,7 +14,7 @@
 from skrl.resources.schedulers.torch import KLAdaptiveLR  # noqa
 from skrl.trainers.torch import SequentialTrainer, Trainer
 from skrl.utils import set_seed
-from skrl.utils.model_instantiators.torch import deterministic_model, gaussian_model, shared_model, categorical_model
+from skrl.utils.model_instantiators.torch import categorical_model, deterministic_model, gaussian_model, shared_model
 
 
 class Runner:
@@ -35,9 +35,9 @@ def __init__(self, env: Union[Wrapper, MultiAgentEnvWrapper], cfg: Mapping[str,
         self._class_mapping = {
             # model
             "gaussianmixin": gaussian_model,
+            "categoricalmixin": categorical_model,
             "deterministicmixin": deterministic_model,
             "shared": shared_model,
-            "categoricalmixin": categorical_model,
             # memory
             "randommemory": RandomMemory,
             # agent