Bunch of CI-demanded fixes.

Author: ludgerpaehler

examples/jaxfluids/test_jaxfluids_env.py

@@ -46,7 +46,7 @@
                 are part of the observation
             - is_scale_observations: Optional. Boolean indicating whether observations are scaled to [0, 1].
             - target_fn: Optional. Target thrust vector function. Choose either 'sine' or 'step'.
-        
+
         The Nozzle3D environment has the following additional arguments:
             - num_actuators: Required. Integer number of actuators. Must be between 4 and 12.
             - secondary_pressure_ratio: Optional. Float, must be between 0.7 and 0.9.
@@ -57,7 +57,7 @@
                 are part of the observation
             - is_scale_observations: Optional. Boolean indicating whether observations are scaled to [0, 1].
             - target_fn: Optional. Target thrust vector function. Choose either 'sine' or 'step'.
-    
+
 """
 
 import os
@@ -77,7 +77,6 @@ def main():
     env.render()
 
     for i in range(1000):
-
         # Random action
         # action = env.action_space.sample()
 
@@ -97,4 +96,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-

hydrogym/jaxfluids/__init__.py

@@ -1 +1 @@
-from .envs.nozzle import Nozzle2D, Nozzle3D
+from .envs.nozzle import Nozzle2D, Nozzle3D

hydrogym/jaxfluids/env_core.py

@@ -34,11 +34,8 @@ class JAXFluidsFlowEnv(JAXFluidsEnv):
     """
 
     def _init_from_hf(self, env_config: dict) -> None:
-
         # Initialize HF data manager
-        self.hf_repo_id = env_config.get(
-            "hf_repo_id", "dynamicslab/HydroGym-environments"
-        )
+        self.hf_repo_id = env_config.get("hf_repo_id", "dynamicslab/HydroGym-environments")
         self.local_fallback_dir = env_config.get("local_fallback_dir", None)
         self.use_clean_cache = env_config.get("use_clean_cache", True)
 
@@ -59,9 +56,7 @@ def _init_from_hf(self, env_config: dict) -> None:
         self.env_data_path = self._setup_environment_data()
 
         # Resolve and load configuration file
-        self.configuration_file = self._resolve_configuration_file(
-            env_config.get("configuration_file")
-        )
+        self.configuration_file = self._resolve_configuration_file(env_config.get("configuration_file"))
 
         if not self.configuration_file:
             raise ConfigError(
@@ -96,13 +91,9 @@ def _setup_environment_data(self) -> str:
             print(f"Using environment data from: {env_path}")
             return env_path
         except Exception as e:
-            raise ConfigError(
-                f"Failed to setup environment data for {self.environment_name}: {e}"
-            )
+            raise ConfigError(f"Failed to setup environment data for {self.environment_name}: {e}")
 
-    def _resolve_configuration_file(
-        self, config_file_input: Optional[str]
-    ) -> Optional[str]:
+    def _resolve_configuration_file(self, config_file_input: Optional[str]) -> Optional[str]:
         """
         Resolve configuration file path from various input formats.
 
@@ -189,9 +180,7 @@ def _find_configuration_file(self) -> Optional[str]:
         for pattern in config_patterns:
             matches = glob.glob(os.path.join(self.env_data_path, pattern))
             if matches:
-                print(
-                    f"Auto-detected configuration file: {os.path.basename(matches[0])}"
-                )
+                print(f"Auto-detected configuration file: {os.path.basename(matches[0])}")
                 return matches[0]
 
         # Not found
@@ -200,4 +189,3 @@ def _find_configuration_file(self) -> Optional[str]:
             print(f"Available files: {os.listdir(self.env_data_path)}")
 
         return None
-