Create test for config_base.py

fixie-ai · Sep 23, 2024 · f942de7 · f942de7
1 parent 8515b3d
commit f942de7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/ultravox/training/config_base_test.py b/ultravox/training/config_base_test.py
@@ -0,0 +1,5 @@
+from ultravox.training import config_base
+
+def test_can_create_train_config():
+    args = config_base.get_train_args()
+    assert isinstance(args, config_base.TrainConfig)