[pre-commit.ci] pre-commit autoupdate (#172)

.pre-commit-config.yaml

@@ -39,7 +39,7 @@ repos:
           - --fix
 
   - repo: https://github.com/psf/black
-    rev: 23.12.1
+    rev: 24.1.1
     hooks:
       - id: black
         additional_dependencies: [toml]

src/stream_mapper/pytorch/_connect/data.py

@@ -58,6 +58,6 @@ def _from_ndarraytype_to_tensor(
         array.flags.writeable = True
         return replace(data, array=xp.asarray(array, **kwargs))
 
-    ASTYPE_REGISTRY[
-        (asdf.tags.core.ndarray.NDArrayType, xp.Tensor)
-    ] = _from_ndarraytype_to_tensor
+    ASTYPE_REGISTRY[(asdf.tags.core.ndarray.NDArrayType, xp.Tensor)] = (
+        _from_ndarraytype_to_tensor
+    )

src/stream_mapper/pytorch/builtin/_skewnorm.py

@@ -87,9 +87,7 @@ def ln_likelihood(
             # TODO: I suspect there are better ways to write this
             sigma_o = data[cens].array[idx]
             sigma = self.xp.exp(ln_s)
-            skew = (
-                skew * sigma / self.xp.sqrt(sigma**2 + (1 + skew**2) * sigma_o**2)
-            )
+            skew = skew * sigma / self.xp.sqrt(sigma**2 + (1 + skew**2) * sigma_o**2)
             ln_s = self.xp.log(sigma**2 + sigma_o**2) / 2
 
         # Find where -inf

src/stream_mapper/pytorch/builtin/_truncskewnorm.py

@@ -80,9 +80,7 @@ def ln_likelihood(
             # TODO: I suspect there are better ways to write this
             sigma_o = data[cens].array[idx]
             sigma = self.xp.exp(ln_s)
-            skew = (
-                skew * sigma / self.xp.sqrt(sigma**2 + (1 + skew**2) * sigma_o**2)
-            )
+            skew = skew * sigma / self.xp.sqrt(sigma**2 + (1 + skew**2) * sigma_o**2)
             ln_s = self.xp.log(sigma**2 + sigma_o**2) / 2
 
         # Find where -inf

src/stream_mapper/pytorch/builtin/compat/nflow.py

@@ -23,7 +23,9 @@ def _log_prob(self, data: Data[Array], idx: Array) -> Array:
         """Log-probability of the array."""
         return self.net.log_prob(
             inputs=data[self.coord_names].array[idx],
-            context=data[self.indep_coord_names].array[idx]
-            if self.indep_coord_names is not None
-            else None,
+            context=(
+                data[self.indep_coord_names].array[idx]
+                if self.indep_coord_names is not None
+                else None
+            ),
         )