Also pyright ignore string types

Author: manuelgloeckler

sbi/samplers/score/diffuser.py

@@ -17,7 +17,7 @@ class Diffuser:
 
     def __init__(
         self,
-        score_based_potential: 'PosteriorScoreBasedPotential',  # noqa: F821
+        score_based_potential: 'PosteriorScoreBasedPotential',  # noqa: F821 # type: ignore
         predictor: Union[str, Predictor],
         corrector: Optional[Union[str, Corrector]] = None,
         predictor_params: Optional[dict] = None,
@@ -62,7 +62,7 @@ def __init__(
     def set_predictor(
         self,
         predictor: Union[str, Predictor],
-        score_based_potential: 'PosteriorScoreBasedPotential',  # noqa: F821
+        score_based_potential: 'PosteriorScoreBasedPotential',  # noqa: F821 # type: ignore
         **kwargs,
     ):
         """Set the predictor for the diffusion-based sampler."""

sbi/samplers/score/predictors.py

@@ -12,7 +12,7 @@
 
 def get_predictor(
     name: str,
-    score_based_potential: 'PosteriorScoreBasedPotential',  # noqa: F821
+    score_based_potential: 'PosteriorScoreBasedPotential',  # noqa: F821 # type: ignore
     **kwargs,
 ) -> "Predictor":
     """Helper function to get predictor by name.
@@ -52,7 +52,7 @@ class Predictor(ABC):
 
     def __init__(
         self,
-        potential_fn: 'PosteriorScoreBasedPotential',  # noqa: F821
+        potential_fn: 'PosteriorScoreBasedPotential',  # noqa: F821 # type: ignore
     ):
         """Initialize predictor.
 
@@ -92,7 +92,7 @@ def predict(self, theta: Tensor, t1: Tensor, t0: Tensor) -> Tensor:
 class EulerMaruyama(Predictor):
     def __init__(
         self,
-        potential_fn: 'PosteriorScoreBasedPotential',  # noqa: F821
+        potential_fn: 'PosteriorScoreBasedPotential',  # noqa: F821 # type: ignore
         eta: float = 1.0,
     ):
         """Simple Euler-Maruyama discretization of the associated family of reverse

sbi/samplers/vi/vi_divergence_optimizers.py

@@ -46,7 +46,7 @@ class DivergenceOptimizer(ABC):
 
     def __init__(
         self,
-        potential_fn: 'BasePotential',  # noqa: F821
+        potential_fn: 'BasePotential',  # noqa: F821 # type: ignore
         q: PyroTransformedDistribution,
         prior: Optional[Distribution] = None,
         n_particles: int = 256,