Fix indentation

stanford-crfm · Feb 11, 2024 · 29538d6 · 29538d6
1 parent 65d0ed0
commit 29538d6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/levanter/data/mixture.py b/src/levanter/data/mixture.py
@@ -40,9 +40,10 @@ def __init__(
     ):
         self.datasets = datasets
         self.weights = MixtureDataset._normalize_weights(weights)
-    
+
         if stop_strategy not in [StopStrategy.FIRST_STOP_STRATEGY, StopStrategy.ALL_STOP_STRATEGY]:
             raise ValueError(f"Stop strategy {stop_strategy} is not supported.")
+
         print(f"=== class MixtureDataset: self.datasets.keys() = {self.datasets.keys()}, self.weights = {self.weights}, stop_strategy = {stop_strategy} ===")
         self.stop_strategy = stop_strategy