Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 2 additions & 6 deletions src/pyrecest/_backend/_shared_numpy/_common.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,5 @@
from pyrecest._backend._dtype_utils import ( # noqa: F401
_np_box_binary_scalar as _box_binary_scalar,
)
from pyrecest._backend._dtype_utils import ( # noqa: F401
_np_box_unary_scalar as _box_unary_scalar,
)
from pyrecest._backend._dtype_utils import _np_box_binary_scalar as _box_binary_scalar # noqa: F401
from pyrecest._backend._dtype_utils import _np_box_unary_scalar as _box_unary_scalar # noqa: F401
from pyrecest._backend._dtype_utils import (
_pre_add_default_dtype_by_casting,
_pre_allow_complex_dtype,
Expand Down
8 changes: 1 addition & 7 deletions src/pyrecest/_backend/autograd/linalg.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,13 +27,7 @@
is_single_matrix_pd,
)
from .._shared_numpy.linalg import logm as _logm
from .._shared_numpy.linalg import (
polar,
qr,
solve,
solve_sylvester,
sqrtm,
)
from .._shared_numpy.linalg import polar, qr, solve, solve_sylvester, sqrtm


def _matrix_transpose(x):
Expand Down
4 changes: 1 addition & 3 deletions src/pyrecest/_backend/numpy/linalg.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,9 +24,7 @@
_normalize_fractional_matrix_power_exponent,
)
from .._shared_numpy.linalg import fractional_matrix_power as _fractional_matrix_power
from .._shared_numpy.linalg import (
is_single_matrix_pd,
)
from .._shared_numpy.linalg import is_single_matrix_pd
from .._shared_numpy.linalg import logm as _logm
from .._shared_numpy.linalg import (
polar,
Expand Down
6 changes: 1 addition & 5 deletions src/pyrecest/_backend/numpy/random.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,7 @@

import numpy as _np
from numpy.random import default_rng as _default_rng
from numpy.random import ( # For PyRecEst
get_state,
seed,
set_state,
)
from numpy.random import get_state, seed, set_state # For PyRecEst

from .._shared_numpy.random import (
_contains_masked_value,
Expand Down
4 changes: 2 additions & 2 deletions src/pyrecest/_backend/pytorch/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
asarray,
)
from torch import broadcast_tensors as broadcast_arrays
from torch import ( # The ones below are for pyrecest; For Riemannian score-based SDE
from torch import (
clip,
column_stack,
complex64,
Expand Down Expand Up @@ -54,7 +54,7 @@
rad2deg,
)
from torch import repeat_interleave as repeat
from torch import ( # The ones below are for pyrecest; For Riemannian score-based SDE
from torch import (
reshape,
roll,
round,
Expand Down
26 changes: 3 additions & 23 deletions src/pyrecest/distributions/abstract_dirac_distribution.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,31 +13,11 @@
all,
)
from pyrecest.backend import any as backend_any
from pyrecest.backend import (
apply_along_axis,
arange,
argmax,
asarray,
)
from pyrecest.backend import apply_along_axis, arange, argmax, asarray
from pyrecest.backend import copy as backend_copy
from pyrecest.backend import (
exp,
int32,
int64,
isclose,
isfinite,
log,
)
from pyrecest.backend import exp, int32, int64, isclose, isfinite, log
from pyrecest.backend import max as backend_max
from pyrecest.backend import (
ones,
random,
reshape,
stack,
sum,
to_numpy,
where,
)
from pyrecest.backend import ones, random, reshape, stack, sum, to_numpy, where

from .abstract_distribution_type import AbstractDistributionType

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,10 +22,7 @@
reshape,
)
from pyrecest.backend import sum as backend_sum
from pyrecest.backend import (
unique,
zeros,
)
from pyrecest.backend import unique, zeros
from pyrecest.distributions.cart_prod.abstract_hypercylindrical_distribution import (
AbstractHypercylindricalDistribution,
)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,9 +9,7 @@
log,
)
from pyrecest.backend import sum as backend_sum
from pyrecest.backend import (
zeros,
)
from pyrecest.backend import zeros

from ..nonperiodic.gaussian_distribution import GaussianDistribution
from ..nonperiodic.gaussian_mixture import GaussianMixture
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -6,17 +6,9 @@
allclose,
)
from pyrecest.backend import any as backend_any
from pyrecest.backend import (
argmax,
array,
asarray,
concatenate,
stack,
)
from pyrecest.backend import argmax, array, asarray, concatenate, stack
from pyrecest.backend import sum as backend_sum
from pyrecest.backend import (
zeros,
)
from pyrecest.backend import zeros

from ..nonperiodic.gaussian_distribution import GaussianDistribution
from ..nonperiodic.gaussian_mixture import GaussianMixture
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,15 +12,7 @@
log,
)
from pyrecest.backend import max as backend_max
from pyrecest.backend import (
mean,
mod,
pi,
random,
sqrt,
sum,
zeros,
)
from pyrecest.backend import mean, mod, pi, random, sqrt, sum, zeros

from .abstract_circular_distribution import AbstractCircularDistribution

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,13 +7,7 @@
asarray,
)
from pyrecest.backend import copy as backend_copy
from pyrecest.backend import (
exp,
isfinite,
mod,
pi,
to_numpy,
)
from pyrecest.backend import exp, isfinite, mod, pi, to_numpy

from .abstract_circular_distribution import AbstractCircularDistribution

Expand Down
4 changes: 1 addition & 3 deletions src/pyrecest/distributions/conversion.py
Original file line number Diff line number Diff line change
Expand Up @@ -797,9 +797,7 @@ def _validate_conversion_arguments(
)


from . import ( # noqa: F401,E402 pylint: disable=wrong-import-position,unused-import
so3_conversion as _so3_conversion,
)
from . import so3_conversion as _so3_conversion # noqa: F401,E402 pylint: disable=wrong-import-position,unused-import

__all__ = [
"ConversionError",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,12 +13,7 @@
pi,
)
from pyrecest.backend import random as backend_random
from pyrecest.backend import (
sin,
sqrt,
stack,
where,
)
from pyrecest.backend import sin, sqrt, stack, where

from .abstract_hypersphere_subset_uniform_distribution import (
AbstractHypersphereSubsetUniformDistribution,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,7 @@

# pylint: disable=no-name-in-module,no-member
from pyrecest.backend import all as backend_all
from pyrecest.backend import (
allclose,
)
from pyrecest.backend import allclose
from pyrecest.backend import copy as backend_copy
from pyrecest.backend import (
exp,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,15 +8,7 @@

# pylint: disable=no-name-in-module,no-member
from pyrecest.backend import all as backend_all
from pyrecest.backend import (
asarray,
cov,
isfinite,
ones,
reshape,
to_numpy,
zeros,
)
from pyrecest.backend import asarray, cov, isfinite, ones, reshape, to_numpy, zeros

from ..abstract_dirac_distribution import AbstractDiracDistribution
from .abstract_linear_distribution import AbstractLinearDistribution
Expand Down
14 changes: 2 additions & 12 deletions src/pyrecest/filters/bingham_filter.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,19 +3,9 @@

import pyrecest.backend
from pyrecest.backend import all as backend_all
from pyrecest.backend import (
allclose,
array,
asarray,
)
from pyrecest.backend import allclose, array, asarray
from pyrecest.backend import copy as backend_copy
from pyrecest.backend import (
diag,
eye,
isclose,
isfinite,
linalg,
)
from pyrecest.backend import diag, eye, isclose, isfinite, linalg
from pyrecest.distributions.hypersphere_subset.bingham_distribution import (
BinghamDistribution,
)
Expand Down
12 changes: 2 additions & 10 deletions src/pyrecest/filters/goal_conditioned_replay_imm_filter.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,17 +41,9 @@
log,
)
from pyrecest.backend import max as backend_max
from pyrecest.backend import (
maximum,
ndim,
outer,
pi,
reshape,
)
from pyrecest.backend import maximum, ndim, outer, pi, reshape
from pyrecest.backend import sum as backend_sum
from pyrecest.backend import (
zeros,
)
from pyrecest.backend import zeros
from pyrecest.distributions import GaussianDistribution, LinearDiracDistribution

from .abstract_filter import AbstractFilter
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,11 +14,7 @@
atleast_2d,
)
from pyrecest.backend import copy as backend_copy
from pyrecest.backend import (
eye,
linalg,
zeros,
)
from pyrecest.backend import eye, linalg, zeros
from pyrecest.distributions import GaussianDistribution
from pyrecest.protocols.common import BackendArray

Expand Down
4 changes: 1 addition & 3 deletions src/pyrecest/filters/lomem_tracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,9 +19,7 @@
sin,
)
from pyrecest.backend import sum as backend_sum
from pyrecest.backend import (
zeros,
)
from pyrecest.backend import zeros

from .abstract_extended_object_tracker import AbstractExtendedObjectTracker

Expand Down
6 changes: 1 addition & 5 deletions src/pyrecest/filters/mem_rbpf_tracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,11 +38,7 @@
stack,
)
from pyrecest.backend import sum as backend_sum
from pyrecest.backend import (
transpose,
where,
zeros,
)
from pyrecest.backend import transpose, where, zeros
from pyrecest.numerics import assert_covariance_matrix

from .abstract_extended_object_tracker import AbstractExtendedObjectTracker
Expand Down
10 changes: 1 addition & 9 deletions src/pyrecest/filters/multi_bernoulli_tracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,15 +7,7 @@

# pylint: disable=no-name-in-module,no-member
from pyrecest.backend import all as backend_all
from pyrecest.backend import (
argmax,
array,
empty,
full,
linalg,
ndim,
stack,
)
from pyrecest.backend import argmax, array, empty, full, linalg, ndim, stack
from pyrecest.distributions import GaussianDistribution
from scipy.linalg import pinv
from scipy.stats import chi2
Expand Down
16 changes: 2 additions & 14 deletions src/pyrecest/filters/multi_hypothesis_tracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,22 +6,10 @@
from operator import index as operator_index

from pyrecest.backend import all as backend_all
from pyrecest.backend import (
argmax,
array,
asarray,
exp,
full,
linalg,
)
from pyrecest.backend import argmax, array, asarray, exp, full, linalg
from pyrecest.backend import log as backend_log
from pyrecest.backend import max as backend_max
from pyrecest.backend import (
ndim,
pi,
sqrt,
stack,
)
from pyrecest.backend import ndim, pi, sqrt, stack
from pyrecest.backend import sum as backend_sum
from pyrecest.distributions import GaussianDistribution
from pyrecest.utils import murty_k_best_assignments
Expand Down
8 changes: 1 addition & 7 deletions src/pyrecest/filters/out_of_sequence.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,13 +19,7 @@
float64,
)
from pyrecest.backend import isfinite as backend_isfinite
from pyrecest.backend import (
linalg,
maximum,
minimum,
transpose,
where,
)
from pyrecest.backend import linalg, maximum, minimum, transpose, where
from pyrecest.distributions import GaussianDistribution

from .kalman_filter import KalmanFilter
Expand Down
9 changes: 2 additions & 7 deletions src/pyrecest/filters/relaxed_s3f_circular.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,9 +18,7 @@

import numpy as np
from pyrecest.backend import abs as backend_abs
from pyrecest.backend import (
all,
)
from pyrecest.backend import all
from pyrecest.backend import angle as backend_angle
from pyrecest.backend import (
array,
Expand All @@ -37,10 +35,7 @@
stack,
)
from pyrecest.backend import sum as backend_sum
from pyrecest.backend import (
zeros,
zeros_like,
)
from pyrecest.backend import zeros, zeros_like
from pyrecest.exceptions import NumericalStabilityError, ShapeError
from pyrecest.numerics import assert_covariance_matrix

Expand Down
Loading