diff --git a/slate/slate.py b/slate/slate.py index 5ee69a4..3d10a8a 100644 --- a/slate/slate.py +++ b/slate/slate.py @@ -223,7 +223,12 @@ class Slate(): ) as run: runner(self, run, config) - assert not REQUIRE_CONFIG_CONSUMED or config == {}, ('Config was not completely consumed: ', config) + if config != {}: + msg = ('Config was not completely consumed: ', config) + if REQUIRE_CONFIG_CONSUMED: + raise Exception(msg) + else: + print(msg) orig_config = config def _run_from_sweep(self, orig_config, p_ind): @@ -240,7 +245,12 @@ class Slate(): self.deep_update(config, wandb.config) runner(self, run, config) - assert config == {}, ('Config was not completely consumed: ', config) + if config != {}: + msg = ('Config was not completely consumed: ', config) + if REQUIRE_CONFIG_CONSUMED: + raise Exception(msg) + else: + print(msg) orig_config = config def from_args(self):