f

ko3n1g · ko3n1g · commit 4a350be18bc5 · 2025-03-05T23:35:35.000+01:00
Signed-off-by: oliver könig &lt;okoenig@nvidia.com&gt;
diff --git a/nemo_run/__init__.py b/nemo_run/__init__.py
@@ -31,7 +31,7 @@
 from nemo_run.devspace.base import DevSpace
 from nemo_run.help import help
 from nemo_run.lazy import LazyEntrypoint, lazy_imports
-from nemo_run.package_info import __version__
+from nemo_run.package_info import __package_name__, __version__
 from nemo_run.run.api import run
 from nemo_run.run.experiment import Experiment
 from nemo_run.run.plugin import ExperimentPlugin as Plugin
@@ -70,5 +70,6 @@
     "Torchrun",
     "SlurmRay",
     "SlurmTemplate",
-    "__version__"
+    "__version__",
+    "__package_name__",
 ]
diff --git a/nemo_run/package_info.py b/nemo_run/package_info.py
@@ -29,3 +29,4 @@
 VERSION = (MAJOR, MINOR, PATCH, PRE_RELEASE, DEV)
 
 __shortversion__ = ".".join(map(str, VERSION[:3]))
+__package_name__ = "nemo_run"

Original file line number	Diff line number	Diff line change
`@@ -29,3 +29,4 @@`
`29`	`29`	`VERSION = (MAJOR, MINOR, PATCH, PRE_RELEASE, DEV)`
`30`	`30`
`31`	`31`	`__shortversion__ = ".".join(map(str, VERSION[:3]))`
	`32`	`+__package_name__ = "nemo_run"`