Fix sdk references

Signed-off-by: Ryan Wolf <[email protected]>
NVIDIA · Aug 6, 2024 · 2b04fca · 2b04fca
1 parent 59e9145
commit 2b04fca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/nemo_run/launch_slurm.py b/examples/nemo_run/launch_slurm.py
@@ -46,8 +46,8 @@ def main():
         script_command=curator_command,
     )
 
-    executor = sdk.resolve(SlurmExecutor, "nemo_curator_slurm_executor")
-    with sdk.Experiment("example_nemo_curator_exp", executor=executor) as exp:
+    executor = run.resolve(SlurmExecutor, "nemo_curator_slurm_executor")
+    with run.Experiment("example_nemo_curator_exp", executor=executor) as exp:
         exp.add(curator_job.to_script(), tail_logs=True)
         exp.run(detach=False)