Merge pull request #191 from AstroAI-Lab/main

update joss branch

Author: TobiBu

.pre-commit-config.yaml

@@ -1,7 +1,7 @@
 
 repos:
   - repo: https://github.com/kynan/nbstripout
-    rev: 0.8.2
+    rev: 0.9.0
     hooks:
     - id: nbstripout
       files: ".ipynb"
@@ -14,7 +14,7 @@ repos:
     -   id: check-merge-conflict # Check for files that contain merge conflict strings.
     -   id: check-added-large-files # Prevent large files from being added to the repository.
   - repo: https://github.com/psf/black-pre-commit-mirror
-    rev: 25.12.0
+    rev: 26.1.0
     hooks:
     - id: black
   - repo: https://github.com/pycqa/isort

rubix/core/data.py

@@ -415,7 +415,7 @@ def prepare_input(config: Union[dict, str]) -> RubixData:
             >>> rubixdata.stars.coords
     """
 
-    logger_config = config["logger"] if "logger" in config else None  # type:ignore
+    logger_config = config["logger"] if "logger" in config else None  # type: ignore
     logger = get_logger(logger_config)
     file_path = config["output_path"]
     file_path = os.path.join(file_path, "rubix_galaxy.h5")
@@ -464,13 +464,13 @@ def prepare_input(config: Union[dict, str]) -> RubixData:
                 if rubixdata.stars.coords is not None:
                     indices = np.random.choice(
                         np.arange(len(rubixdata.stars.coords)),
-                        size=size,  # type:ignore
+                        size=size,  # type: ignore
                         replace=False,
-                    )  # type:ignore
+                    )  # type: ignore
                 elif rubixdata.gas.coords is not None:
                     indices = np.random.choice(
                         np.arange(len(rubixdata.gas.coords)),
-                        size=size,  # type:ignore
+                        size=size,  # type: ignore
                         replace=False,
                     )
                 else:

rubix/logger.py

@@ -45,15 +45,13 @@ def get_logger(config=None):
 
     if first_time:
 
-        logger.info(
-            r"""
+        logger.info(r"""
    ___  __  _____  _____  __
   / _ \/ / / / _ )/  _/ |/_/
  / , _/ /_/ / _  |/ /_>  <
 /_/|_|\____/____/___/_/|_|
 
-"""
-        )
+""")
         logger.info(f"Rubix version: {version.__version__}")
         logger.info(f"JAX version: {jax.__version__}")
         logger.info(f"Running on {jax.devices()} devices")

tests/test_illustris_api.py

@@ -385,11 +385,11 @@ def test_append_subhalo_data(api_instance, tmp_path):
                 dataset = f["SubhaloData"][key]  # type: ignore
                 if isinstance(value, np.ndarray):
                     np.testing.assert_array_equal(
-                        dataset[:], value  # type:ignore
-                    ), f"Data '{key}' does not match the expected value."  # type:ignore
+                        dataset[:], value  # type: ignore
+                    ), f"Data '{key}' does not match the expected value."  # type: ignore
                 else:
                     assert (
-                        dataset[()] == value  # type:ignore
+                        dataset[()] == value  # type: ignore
                     ), f"Data '{key}' does not match the expected value."