Add support for YAML, TOML & JSON (#182)

Author: marcromeyn

docs/source/guides/cli.md

@@ -69,5 +69,16 @@ def train_model(
     print("Training completed!")
 
 
+@run.cli.factory(target=train_model)
+def train_recipe() -> run.Partial[train_model]:
+    return run.Partial(
+        train_model,
+        model=my_model(hidden_size=512),
+        optimizer=my_optimizer(learning_rate=0.0005),
+        epochs=50,
+        batch_size=2048,
+    )
+
+
 if __name__ == "__main__":
     run.cli.main(train_model, cmd_defaults={"skip_confirmation": True})

examples/entrypoint/task.py

@@ -0,0 +1,16 @@
+_partial_ = true
+_target_ = "__main__.train_model"
+batch_size = 32
+epochs = 10
+
+[model]
+_target_ = "__main__.Model"
+activation = "relu"
+hidden_size = 256
+num_layers = 5
+
+[optimizer]
+_target_ = "__main__.Optimizer"
+betas = [ 0.9, 0.999,]
+learning_rate = 0.001
+weight_decay = 1e-5

examples/entrypoint/test.toml

@@ -26,7 +26,7 @@
 from nemo_run.core.tunnel.client import LocalTunnel, SSHTunnel
 from nemo_run.devspace.base import DevSpace
 from nemo_run.help import help
-from nemo_run.lazy import LazyEntrypoint, lazy_imports
+from nemo_run.cli.lazy import LazyEntrypoint, lazy_imports
 from nemo_run.package_info import __package_name__, __version__
 from nemo_run.run.api import run
 from nemo_run.run.experiment import Experiment