Fix formatting issues in walker.py and domain_randomization.py

SuReLI · Apr 24, 2024 · e875c0d · e875c0d
1 parent b1ae360
commit e875c0d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/rrls/envs/walker.py b/rrls/envs/walker.py
@@ -114,7 +114,9 @@ def set_params(
             else getattr(self, "thighmass", DEFAULT_PARAMS["thighmass"])
         )
         self.legmass = (
-            legmass if legmass is not None else getattr(self, "legmass", DEFAULT_PARAMS["legmass"])
+            legmass
+            if legmass is not None
+            else getattr(self, "legmass", DEFAULT_PARAMS["legmass"])
         )
         self.footmass = (
             footmass

diff --git a/rrls/wrappers/domain_randomization.py b/rrls/wrappers/domain_randomization.py
@@ -30,10 +30,9 @@ def __init__(
         self,
         env: ModifiedParamsEnv,
         params_bound: dict[str, Annotated[tuple[float], 2]],
-        randomize_fn: Callable[
-            [dict[str, Annotated[tuple[float], 2]]], dict[str, float]
-        ]
-        | None = None,
+        randomize_fn: (
+            Callable[[dict[str, Annotated[tuple[float], 2]]], dict[str, float]] | None
+        ) = None,
     ):
         super().__init__(env)
         self.env = env