feat: fixed pyright complaints with SilentTQDM and made not printing default behaviour

CHANGELOG.md

@@ -13,6 +13,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Added `coreax.kernels.PowerKernel` to replace repeated calls of `coreax.kernels.ProductKernel`
 within the `**` magic method of `coreax.kernel.ScalarValuedKernel`
 - Added scalar-valued kernel functions `coreax.kernels.PoissonKernel` and `coreax.kernels.MaternKernel`
+- Added `progress_bar` attribute to `coreax.score_matching.SlicedScoreMatching` to enable or
+disable tqdm progress bar terminal output. Defaults to `` :data:`False` ``.
 
 
 ### Fixed

coreax/score_matching.py

@@ -146,7 +146,7 @@ def __init__(  # noqa: PLR0913, PLR0917
         num_noise_models: int = 100,
         sigma: float = 1.0,
         gamma: float = 0.95,
-        print_progress_bar: bool = False,
+        progress_bar: bool = False,
     ):
         """Define a sliced score matching class and update invalid inputs."""
         # JAX will not error if we have num_random_vectors set to 0, but this approach
@@ -170,7 +170,7 @@ def __init__(  # noqa: PLR0913, PLR0917
         self.num_noise_models = num_noise_models
         self.sigma = sigma
         self.gamma = gamma
-        if print_progress_bar:
+        if progress_bar:
             self.progress_bar = LoudTQDM
         else:
             self.progress_bar = SilentTQDM

coreax/util.py

@@ -467,4 +467,3 @@ def __iter__(self) -> Iterator[T]:
     @staticmethod
     def write(*args, **kwargs) -> None:
         """Do nothing instead of writing to output."""
-        del args, kwargs

documentation/source/conf.py

@@ -196,7 +196,6 @@
     ("py:obj", "coreax.weights._Data"),
     ("py:obj", "coreax.metrics._Data"),
     ("py:obj", "coreax.solvers.coresubset._SupervisedData"),
-    ("py:obj", "coreax.solvers.coresubset._SupervisedData"),
 ]