From 29538d6024254ea34c851c271040da35992514df Mon Sep 17 00:00:00 2001 From: Ivan Zhou Date: Sun, 11 Feb 2024 02:17:36 +0000 Subject: [PATCH] Fix indentation --- src/levanter/data/mixture.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/levanter/data/mixture.py b/src/levanter/data/mixture.py index 23f2ba37d..3612a27bd 100644 --- 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