Add option to disable progress bar

autoemulate/experimental/learners/stream.py

@@ -1,5 +1,5 @@
 from abc import abstractmethod
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 
 import numpy as np
 import torch
@@ -19,6 +19,8 @@ class Stream(Active):
     (Inherits parameters from Active)
     """
 
+    show_progress: bool = field(default=True)
+
     @abstractmethod
     def query(
         self, X: TensorLike | None = None
@@ -51,7 +53,14 @@ def fit_samples(self, X: torch.Tensor):
             Stream of input samples.
         """
         X = self.check_matrix(X)
-        for x in (pb := tqdm(X, desc=self.__class__.__name__, leave=True)):
+        for x in (
+            pb := tqdm(
+                X,
+                desc=self.__class__.__name__,
+                leave=True,
+                disable=not self.show_progress,
+            )
+        ):
             self.fit(x.reshape(1, -1))
             pb.set_postfix(
                 ordered_dict={key: val[-1] for key, val in self.metrics.items()}
@@ -76,6 +85,7 @@ def fit_batches(self, X: torch.Tensor, batch_size: int):
             pb := tqdm(
                 range(0, X.shape[0], batch_size),
                 desc=f"{self.__class__.__name__} (batches)",
+                disable=not self.show_progress,
             )
         ):
             batch = X[i : i + batch_size]