[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

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."