[Fix] Fix ruff errors and a few type annotations

This commit is contained in:
Fabian Joswig 2024-12-25 11:44:24 +01:00
parent 8d86295ac5
commit 2d34b355ed
5 changed files with 16 additions and 6 deletions

View file

@ -7,7 +7,7 @@ import matplotlib.pyplot as plt
import scipy.linalg import scipy.linalg
from .obs import Obs, reweight, correlate, CObs from .obs import Obs, reweight, correlate, CObs
from .misc import dump_object, _assert_equal_properties from .misc import dump_object, _assert_equal_properties
from .fits import least_squares from .fits import least_squares, Fit_result
from .roots import find_root from .roots import find_root
from . import linalg from . import linalg
from numpy import float64, int64, ndarray, ufunc from numpy import float64, int64, ndarray, ufunc

View file

@ -6,11 +6,13 @@ import matplotlib
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import pandas as pd import pandas as pd
import pickle import pickle
from .obs import Obs from .obs import Obs, CObs
from .version import __version__ from .version import __version__
from numpy import float64, int64, ndarray from numpy import float64, int64, ndarray
from typing import List, Type, Union from typing import List, Type, Union, TYPE_CHECKING
if TYPE_CHECKING:
from .correlators import Corr
def print_config(): def print_config():
"""Print information about version of python, pyerrors and dependencies.""" """Print information about version of python, pyerrors and dependencies."""

View file

@ -3,7 +3,6 @@ import numpy as np
import scipy.linalg import scipy.linalg
from .obs import Obs from .obs import Obs
from .linalg import svd, eig from .linalg import svd, eig
from pyerrors.obs import Obs
from typing import List from typing import List

View file

@ -1,4 +1,5 @@
from __future__ import annotations from __future__ import annotations
import sys
import warnings import warnings
import hashlib import hashlib
import pickle import pickle
@ -12,7 +13,15 @@ import numdifftools as nd
from itertools import groupby from itertools import groupby
from .covobs import Covobs from .covobs import Covobs
from numpy import bool, float64, int64, ndarray from numpy import bool, float64, int64, ndarray
from typing import Any, Callable, Dict, List, Optional, Union from typing import Any, Callable, Dict, List, Optional, Union, TYPE_CHECKING
if sys.version_info >= (3, 10):
from types import NotImplementedType
else:
NotImplementedType = type(NotImplemented)
if TYPE_CHECKING:
from .correlators import Corr
# Improve print output of numpy.ndarrays containing Obs objects. # Improve print output of numpy.ndarrays containing Obs objects.
np.set_printoptions(formatter={'object': lambda x: str(x)}) np.set_printoptions(formatter={'object': lambda x: str(x)})

View file

@ -2,7 +2,7 @@ from __future__ import annotations
import numpy as np import numpy as np
import scipy.optimize import scipy.optimize
from autograd import jacobian from autograd import jacobian
from .obs import derived_observable from .obs import Obs, derived_observable
from typing import Callable, List, Union from typing import Callable, List, Union