Adding more tests

Signed-off-by: Marc Romeyn <mromeijn@nvidia.com>

Author: marcromeyn

nemo_run/__init__.py

@@ -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

nemo_run/cli/api.py

@@ -67,8 +67,8 @@
 from nemo_run.core.execution import LocalExecutor, SkypilotExecutor, SlurmExecutor
 from nemo_run.core.execution.base import Executor
 from nemo_run.core.frontend.console.styles import BOX_STYLE, TABLE_STYLES
-from nemo_run.core.serialization.yaml import ConfigSerializer
-from nemo_run.lazy import LazyEntrypoint
+from nemo_run.cli.config import ConfigSerializer
+from nemo_run.cli.lazy import LazyEntrypoint
 from nemo_run.run.experiment import Experiment
 from nemo_run.run.plugin import ExperimentPlugin as Plugin
 

nemo_run/cli/cli_parser.py

@@ -1046,6 +1046,11 @@ def dummy_model_config():
         key, (op, value) = next(iter(parsed.items()))
         logger.debug(f"Parsed key: {key}, op: {op}, value: {value}")
 
+        # Skip the internal _target_ argument used for serialization
+        if key == "_target_":
+            logger.debug(f"Skipping internal '_target_' argument: {arg}")
+            continue
+
         if "." not in key:
             if isinstance(fn, (Config, Partial)):
                 signature = inspect.signature(fn.__fn_or_cls__)