Create error field script

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 repos:
 -   repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v3.4.0
+    rev: v4.0.1
     hooks:
     - id: check-added-large-files
     - id: check-case-conflict
@@ -17,7 +17,7 @@ repos:
     - id: trailing-whitespace
 
 -   repo: https://github.com/pre-commit/pygrep-hooks
-    rev: v1.7.0
+    rev: v1.9.0
     hooks:
     - id: python-check-blanket-noqa
     - id: python-check-mock-methods
@@ -29,12 +29,12 @@ repos:
     - id: text-unicode-replacement-char
 
 -   repo: https://github.com/asottile/add-trailing-comma
-    rev: v2.0.2
+    rev: v2.1.0
     hooks:
     -   id: add-trailing-comma
 
 -   repo: https://github.com/jumanjihouse/pre-commit-hooks
-    rev: 2.1.4
+    rev: 2.1.5
     hooks:
     - id: check-mailmap
     - id: forbid-binary
@@ -43,12 +43,12 @@ repos:
     - id: shfmt
 
 -   repo: https://github.com/mgedmin/check-manifest
-    rev: "0.46"
+    rev: '0.46'
     hooks:
     - id: check-manifest
 
 -   repo: https://github.com/pre-commit/mirrors-autopep8
-    rev: v1.5.4
+    rev: v1.5.7
     hooks:
     - id: autopep8
 
@@ -57,13 +57,13 @@ repos:
     hooks:
     - id: seed-isort-config
 -   repo: https://github.com/pre-commit/mirrors-isort
-    rev: v5.7.0
+    rev: v5.9.3
     hooks:
     - id: isort
       additional_dependencies: ["toml"]
 
 -   repo: https://gitlab.com/pycqa/flake8
-    rev: 3.8.4
+    rev: 3.9.2
     hooks:
     - id: flake8
       args: # arguments to configure flake8
@@ -79,17 +79,17 @@ repos:
         - "--exclude=*/_astropy_init.py docs/conf.py"
 
 -   repo: https://github.com/psf/black
-    rev: 20.8b1
+    rev: 21.8b0
     hooks:
     - id: black
       additional_dependencies: ["toml"]
 -   repo: https://github.com/asottile/blacken-docs
-    rev: v1.9.1
+    rev: v1.11.0
     hooks:
     -   id: blacken-docs
         additional_dependencies: [black==20.8b1]
 - repo: https://github.com/nbQA-dev/nbQA
-  rev: 0.5.6
+  rev: 1.1.0
   hooks:
     - id: nbqa-black
       additional_dependencies: [black==20.8b1]

discO/_astropy_init.py

@@ -63,8 +63,7 @@
                     update_default_config(__package__, config_dir)
                 except ConfigurationDefaultMissingError as e:
                     wmsg = (
-                        e.args[0]
-                        + " Cannot install default profile. If you are "
+                        e.args[0] + " Cannot install default profile. If you are "
                         "importing from source, this is expected."
                     )
                     warn(ConfigurationDefaultMissingWarning(wmsg))

discO/conftest.py

@@ -22,10 +22,7 @@
 
 try:
     # THIRD PARTY
-    from pytest_astropy_header.display import (
-        PYTEST_HEADER_MODULES,
-        TESTED_VERSIONS,
-    )
+    from pytest_astropy_header.display import PYTEST_HEADER_MODULES, TESTED_VERSIONS
 
     ASTROPY_HEADER = True
 except ImportError:

discO/core/common.py

@@ -137,11 +137,7 @@ def registry(cls) -> T.Mapping:
 
         # else, filter registry by subclass
         return MappingProxyType(
-            {
-                k: v
-                for k, v in cls._registry.items()
-                if issubclass(v, cls) and v is not cls
-            },
+            {k: v for k, v in cls._registry.items() if issubclass(v, cls) and v is not cls},
         )
 
     # /def

discO/core/fitter.py

@@ -28,10 +28,7 @@
 # PROJECT-SPECIFIC
 import discO.type_hints as TH
 from .common import CommonBase
-from discO.utils.coordinates import (
-    resolve_framelike,
-    resolve_representationlike,
-)
+from discO.utils.coordinates import resolve_framelike, resolve_representationlike
 from discO.utils.pbar import get_progress_bar
 
 ##############################################################################
@@ -114,16 +111,13 @@ def __new__(
 
             if key not in cls._registry:
                 raise ValueError(
-                    "PotentialFitter has no registered fitter for key: "
-                    f"{key}",
+                    "PotentialFitter has no registered fitter for key: " f"{key}",
                 )
 
             # from registry. Registered in __init_subclass__
             kls = cls._registry[key]
             kwargs.pop("key", None)  # it's already used.
-            return kls.__new__(
-                kls, potential_cls=potential_cls, key=None, **kwargs
-            )
+            return kls.__new__(kls, potential_cls=potential_cls, key=None, **kwargs)
 
         elif key is not None:
             raise ValueError(

discO/core/measurement.py

@@ -37,12 +37,7 @@
 import astropy.units as u
 import numpy as np
 import scipy.stats
-from astropy.coordinates import (
-    BaseCoordinateFrame,
-    BaseRepresentation,
-    SkyCoord,
-    concatenate,
-)
+from astropy.coordinates import BaseCoordinateFrame, BaseRepresentation, SkyCoord, concatenate
 from astropy.utils.decorators import classproperty
 
 # PROJECT-SPECIFIC
@@ -173,8 +168,7 @@ def __new__(
 
         elif method is not None:
             raise ValueError(
-                f"Can't specify 'method' on {cls},"
-                " only on MeasurementErrorSampler.",
+                f"Can't specify 'method' on {cls}," " only on MeasurementErrorSampler.",
             )
 
         return super().__new__(cls)
@@ -398,9 +392,7 @@ def _run_batch(
         else:  # (Nsamples, Niter)
 
             samples = list(
-                self._run_iter(
-                    c, c_err=c_err, random=random, progress=progress, **kwargs
-                ),
+                self._run_iter(c, c_err=c_err, random=random, progress=progress, **kwargs),
             )
 
             sample = concatenate(samples).reshape(c.shape)
@@ -457,9 +449,7 @@ def run(
         if not isinstance(random, np.random.RandomState):
             random = np.random.RandomState(random)
 
-        return run_func(
-            c, c_err=c_err, random=random, progress=progress, **kwargs
-        )
+        return run_func(c, c_err=c_err, random=random, progress=progress, **kwargs)
 
     # /def
 
@@ -698,15 +688,12 @@ def __new__(
             # a cleaner error than KeyError on the actual registry
             if not cls._in_registry(method):
                 raise ValueError(
-                    "RVS_Continuous has no registered "
-                    f"measurement resampler '{method}'",
+                    "RVS_Continuous has no registered " f"measurement resampler '{method}'",
                 )
 
             # from registry. Registered in __init_subclass__
             # don't pass rvs, b/c not all subclasses take it
-            return super().__new__(
-                cls[method], c_err=c_err, method=None, **kwargs
-            )
+            return super().__new__(cls[method], c_err=c_err, method=None, **kwargs)
 
         elif method is not None:
             raise ValueError(
@@ -846,10 +833,7 @@ def __call__(
 
         # re-build representation
         new_rep = rep.__class__(
-            **{
-                n: attr_classes[n](p * unit)
-                for p, (n, unit) in zip(new_posT, units.items())
-            }
+            **{n: attr_classes[n](p * unit) for p, (n, unit) in zip(new_posT, units.items())}
         )
         # make coordinate
         new_cc = self.frame.realize_frame(

discO/core/residual.py

@@ -33,10 +33,7 @@
 import discO.type_hints as TH
 from .common import CommonBase
 from .wrapper import PotentialWrapper
-from discO.utils.coordinates import (
-    resolve_framelike,
-    resolve_representationlike,
-)
+from discO.utils.coordinates import resolve_framelike, resolve_representationlike
 from discO.utils.pbar import get_progress_bar
 
 ##############################################################################
@@ -113,8 +110,7 @@ def __new__(cls, *args, method: T.Optional[str] = None, **kwargs):
 
         elif method is not None:
             raise ValueError(
-                f"Can't specify 'method' on {cls.__name__}, "
-                "only on ResidualMethod.",
+                f"Can't specify 'method' on {cls.__name__}, " "only on ResidualMethod.",
             )
 
         return super().__new__(cls)
@@ -140,9 +136,7 @@ def __init__(
         # representation type
         representation_type = (
             resolve_representationlike(representation_type)
-            if not (
-                representation_type is None or representation_type is Ellipsis
-            )
+            if not (representation_type is None or representation_type is Ellipsis)
             else representation_type
         )
 
@@ -270,9 +264,7 @@ def evaluate_potential(
         evaluator: T.Callable = getattr(evaluator_cls, observable)
 
         # evaluate
-        value = evaluator(
-            points, representation_type=representation_type, **kwargs
-        )
+        value = evaluator(points, representation_type=representation_type, **kwargs)
 
         # -----------------------
         # Return

discO/core/sample.py

@@ -154,8 +154,7 @@ def __new__(
 
             if key not in cls._registry:
                 raise ValueError(
-                    "PotentialSampler has no registered sampler for key: "
-                    f"{key}",
+                    "PotentialSampler has no registered sampler for key: " f"{key}",
                 )
 
             # from registry. Registered in __init_subclass__
@@ -195,8 +194,7 @@ def __init__(
         mtot = potential.total_mass() if total_mass is None else total_mass
         if not np.isfinite(mtot):  # divergent
             raise ValueError(
-                "The potential`s mass is divergent, "
-                "the argument `total_mass` cannot be None.",
+                "The potential`s mass is divergent, " "the argument `total_mass` cannot be None.",
             )
 
         # potential is checked in __new__ as a PotentialWrapper
@@ -642,19 +640,16 @@ def _get_dimensions(meshgrid):
     def _imapper(self):
         @frompyfunc(nin=1, nout=1)
         def imapper(uniform_draw):
-            iflat = np.where(
-                uniform_draw * self._normalization <= self._index_partition,
-            )[0][0]
+            iflat = np.where(uniform_draw * self._normalization <= self._index_partition)
+            iflat = iflat[0][0]
             i = np.unravel_index(iflat, self._gridshape)
             return i
 
         return imapper
 
     # /def
 
-    def __call__(
-        self, n: int, rng: T.Optional[np.random.Generator] = None, **kw
-    ):
+    def __call__(self, n: int, rng: T.Optional[np.random.Generator] = None, **kw):
         """Sample.
 
         .. todo::

discO/core/tests/test_fitter.py

@@ -323,10 +323,7 @@ def test_representation_type(self):
     def test_potential_kwargs(self):
         """Test attribute ``potential_kwargs``."""
         if hasattr(self.inst, "_instance"):
-            assert (
-                self.inst.potential_kwargs
-                == self.inst._instance.potential_kwargs
-            )
+            assert self.inst.potential_kwargs == self.inst._instance.potential_kwargs
         else:
             assert self.inst.potential_kwargs == MappingProxyType(
                 self.inst._kwargs,
@@ -388,10 +385,7 @@ def test_run(self, sample, mass, batch):
             assert isinstance(pots, np.ndarray)
             assert len(pots) == sample.shape[1]
             assert all(
-                [
-                    isinstance(p.__wrapped__, sample.cache["potential"])
-                    for p in pots
-                ],
+                [isinstance(p.__wrapped__, sample.cache["potential"]) for p in pots],
             )
 
         # and then cleanup