more clean up

allenai · Nov 26, 2024 · 4ef79a7 · 4ef79a7
1 parent d5843c1
commit 4ef79a7
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 12 deletions.
diff --git a/src/olmo_core/internal/model_ladder.py b/src/olmo_core/internal/model_ladder.py
@@ -157,6 +157,7 @@ def main(ladder_builder: Callable[[str], ModelLadder]):
 
     # Build run config.
     config = build_config(ladder, script, size, cmd, cluster, overrides)
+    config.ladder.validate()
 
     # Run the cmd.
     cmd.run(size, config)
diff --git a/src/test/internal/__init__.py b/src/test/internal/__init__.py
diff --git a/src/test/internal/model_ladder_test.py b/src/test/internal/model_ladder_test.py
diff --git a/src/test/model_ladder_test.py b/src/test/model_ladder_test.py
@@ -1,4 +1,4 @@
-from olmo_core.ladder import ModelSize
+from olmo_core.model_ladder import ModelSize
 
 
 def test_model_size_num_params():