Update to latest Lepton SDK (#248)

The Lepton SDK has been updated for compatibility with DGX Cloud Lepton.
Existing installations of the Lepton SDK will need to be updated to work
with DGX Cloud Lepton.

Signed-off-by: Robert Clark <[email protected]>

Author: roclark

nemo_run/core/execution/lepton.py

@@ -11,7 +11,7 @@
 from typing import Any, List, Optional, Set, Type
 
 from invoke.context import Context
-from leptonai.api.v1.client import APIClient
+from leptonai.api.v2.client import APIClient
 from leptonai.api.v1.types.affinity import LeptonResourceAffinity
 from leptonai.api.v1.types.common import LeptonVisibility, Metadata
 from leptonai.api.v1.types.dedicated_node_group import DedicatedNodeGroup

pyproject.toml

@@ -30,7 +30,7 @@ dependencies = [
     "cryptography < 43.0.0",
     "networkx >= 3.3",
     "omegaconf>=2.3.0",
-    "leptonai>=0.23.1",
+    "leptonai>=0.25.0",
     "packaging",
     "toml",
 ]