Fix Codestyle of Jaxfluids merge (#249)

* Format files.

* Resort imports.

* Fix spelling errors.

* Fix linting error.

* Fix linting error.

* Fix formatting.

* Fix linting errors.

* Fix formatting.

Author: ludgerpaehler

examples/jaxfluids/test_jaxfluids_env.py

@@ -8,7 +8,7 @@
 which serves as the common base class.
 
 JAXFluidsFlowEnv has the following arguments:
-    - environment_name: Required. Name of the enviroment.
+    - environment_name: Required. Name of the environment.
     - hf_repo_id: Hugging Face repository (default: 'dynamicslab/HydroGym-environments')
 
     - use_clean_cache: Use clean cache directory (default: True)
@@ -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
@@ -68,7 +68,7 @@
 def main():
     env_config = {
         "environment_name": "Nozzle2D_coarse",
-        "configuration_file": os.path.abspath("environment_config.yaml")
+        "configuration_file": os.path.abspath("environment_config.yaml"),
     }
 
     env = Nozzle2D(env_config=env_config)
@@ -77,12 +77,11 @@ def main():
     env.render()
 
     for i in range(1000):
-
         # Random action
-        # action = env.action_space.sample()       
+        # action = env.action_space.sample()
 
         # Fixed action
-        action = [0.0, 0.5]                  
+        action = [0.0, 0.5]
 
         observation, reward, terminated, truncated, info = env.step(action)
 
@@ -96,4 +95,4 @@ def main():
 
 
 if __name__ == "__main__":
-    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

@@ -3,12 +3,11 @@
 from pathlib import Path
 from typing import Dict, Optional, Tuple, Union
 
+from jaxfluids_rl.jxf_env import JAXFluidsEnv, RenderMode
 from omegaconf import OmegaConf
 
 from hydrogym.data_manager import HFDataManager
 
-from jaxfluids_rl.jxf_env import JAXFluidsEnv, RenderMode
-
 
 class ConfigError(Exception):
     """Exception raised for configuration-related errors."""
@@ -21,7 +20,7 @@ class JAXFluidsFlowEnv(JAXFluidsEnv):
     Base JAXFluidsFlowEnv with Hugging Face Hub integration for configuration management.
 
     Arguments:
-        - environment_name: Required. Name of the enviroment.
+        - environment_name: Required. Name of the environment.
         - hf_repo_id: Hugging Face repository (default: 'dynamicslab/HydroGym-environments')
 
         - use_clean_cache: Use clean cache directory (default: True)
@@ -35,7 +34,6 @@ 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.local_fallback_dir = env_config.get("local_fallback_dir", None)
@@ -45,15 +43,15 @@ def _init_from_hf(self, env_config: dict) -> None:
             repo_id=self.hf_repo_id,
             local_fallback_dir=self.local_fallback_dir,
             use_clean_cache=self.use_clean_cache,
-            fallback_profile="JAXFLUIDS"
+            fallback_profile="JAXFLUIDS",
         )
 
         # Environment identification
         self.environment_name = env_config.get("environment_name")
 
         if not self.environment_name:
             raise ConfigError("'environment_name' must be specified in env_config")
-        
+
         # Download/get environment configuration
         self.env_data_path = self._setup_environment_data()
 
@@ -65,11 +63,10 @@ def _init_from_hf(self, env_config: dict) -> None:
                 f"No configuration file found for environment '{self.environment_name}'. "
                 f"Expected config.yaml in: {self.env_data_path}"
             )
-        
+
         # Load configuration from HF
         self.conf = OmegaConf.load(self.configuration_file)
 
-
     def _setup_environment_data(self) -> str:
         """
         Download and setup environment data from HF Hub.
@@ -95,7 +92,6 @@ def _setup_environment_data(self) -> str:
             return env_path
         except Exception as 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]:
         """
@@ -154,7 +150,7 @@ def _resolve_configuration_file(self, config_file_input: Optional[str]) -> Optio
             f"  - Current directory: {os.getcwd()}\n"
             f"  - Environment directory: {self.env_data_path}"
         )
-    
+
     def _find_configuration_file(self) -> Optional[str]:
         """
         Auto-detect configuration file in the environment data directory.
@@ -192,4 +188,4 @@ def _find_configuration_file(self) -> Optional[str]:
         if os.path.exists(self.env_data_path):
             print(f"Available files: {os.listdir(self.env_data_path)}")
 
-        return None
+        return None