truefoundry 0.11.9__py3-none-any.whl → 0.11.10__py3-none-any.whl

truefoundry/deploy/_autogen/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2025-08-06T19:58:42+00:00
+#   timestamp: 2025-09-09T12:09:08+00:00
 
 from __future__ import annotations
 
@@ -504,34 +504,49 @@ class Profile(str, Enum):
     """
     Name of the MIG profile to use. One of the following based on gpu type
     Please refer to https://docs.nvidia.com/datacenter/tesla/mig-user-guide/#supported-mig-profiles for more details
-    A100 40 GB - [1g.5gb, 1g.10gb, 2g.10gb, 3g.20gb, 4g.20gb]
-    A100 80 GB / H100 80 GB - [1g.10gb, 1g.20gb, 2g.20gb, 3g.40gb, 4g.40gb]
-    H100 94 GB - [1g.12gb, 1g.24gb, 2g.24gb, 3g.47gb, 4g.47gb]
-    H100 96 GB - [1g.12gb, 1g.24gb, 2g.24gb, 3g.48gb, 4g.48gb]
+    A30 - [1g.6gb, 2g.12gb, 4g.24gb]
+    A100 40 GB - [1g.5gb, 1g.10gb, 2g.10gb, 3g.20gb, 4g.20gb, 7g.40gb]
+    A100 80 GB / H100 80 GB - [1g.10gb, 1g.20gb, 2g.20gb, 3g.40gb, 4g.40gb, 7g.80gb]
+    H100 94 GB - [1g.12gb, 1g.24gb, 2g.24gb, 3g.47gb, 4g.47gb, 7g.94gb]
+    H100 96 GB - [1g.12gb, 1g.24gb, 2g.24gb, 3g.48gb, 4g.48gb, 7g.96gb]
     H200 141 GB - [1g.18gb, 1g.35gb, 2g.35gb, 3g.71gb, 4g.71gb]
+    B200 180 GB - [1g.23gb, 1g.45gb, 2g.45gb, 3g.90gb, 4g.90gb, 7g.180gb]
     """
 
+    field_1g_6gb = "1g.6gb"
+    field_2g_12gb = "2g.12gb"
     field_1g_5gb = "1g.5gb"
     field_1g_10gb = "1g.10gb"
-    field_1g_12gb = "1g.12gb"
-    field_1g_18gb = "1g.18gb"
-    field_1g_20gb = "1g.20gb"
-    field_1g_24gb = "1g.24gb"
-    field_1g_35gb = "1g.35gb"
     field_2g_10gb = "2g.10gb"
-    field_2g_20gb = "2g.20gb"
-    field_2g_24gb = "2g.24gb"
-    field_2g_35gb = "2g.35gb"
     field_3g_20gb = "3g.20gb"
-    field_3g_40gb = "3g.40gb"
-    field_3g_47gb = "3g.47gb"
-    field_3g_48gb = "3g.48gb"
-    field_3g_71gb = "3g.71gb"
     field_4g_20gb = "4g.20gb"
+    field_1g_20gb = "1g.20gb"
+    field_2g_20gb = "2g.20gb"
+    field_3g_40gb = "3g.40gb"
     field_4g_40gb = "4g.40gb"
+    field_1g_12gb = "1g.12gb"
+    field_1g_24gb = "1g.24gb"
+    field_2g_24gb = "2g.24gb"
+    field_3g_47gb = "3g.47gb"
     field_4g_47gb = "4g.47gb"
+    field_3g_48gb = "3g.48gb"
     field_4g_48gb = "4g.48gb"
+    field_1g_18gb = "1g.18gb"
+    field_1g_35gb = "1g.35gb"
+    field_2g_35gb = "2g.35gb"
+    field_3g_71gb = "3g.71gb"
     field_4g_71gb = "4g.71gb"
+    field_1g_23gb = "1g.23gb"
+    field_1g_45gb = "1g.45gb"
+    field_2g_45gb = "2g.45gb"
+    field_3g_90gb = "3g.90gb"
+    field_4g_90gb = "4g.90gb"
+    field_4g_24gb = "4g.24gb"
+    field_7g_40gb = "7g.40gb"
+    field_7g_80gb = "7g.80gb"
+    field_7g_94gb = "7g.94gb"
+    field_7g_96gb = "7g.96gb"
+    field_7g_180gb = "7g.180gb"
 
 
 class NvidiaMIGGPU(BaseModel):
@@ -542,7 +557,7 @@ class NvidiaMIGGPU(BaseModel):
     )
     profile: Profile = Field(
         ...,
-        description="Name of the MIG profile to use. One of the following based on gpu type\nPlease refer to https://docs.nvidia.com/datacenter/tesla/mig-user-guide/#supported-mig-profiles for more details\nA100 40 GB - [1g.5gb, 1g.10gb, 2g.10gb, 3g.20gb, 4g.20gb]\nA100 80 GB / H100 80 GB - [1g.10gb, 1g.20gb, 2g.20gb, 3g.40gb, 4g.40gb]\nH100 94 GB - [1g.12gb, 1g.24gb, 2g.24gb, 3g.47gb, 4g.47gb]\nH100 96 GB - [1g.12gb, 1g.24gb, 2g.24gb, 3g.48gb, 4g.48gb]\nH200 141 GB - [1g.18gb, 1g.35gb, 2g.35gb, 3g.71gb, 4g.71gb]",
+        description="Name of the MIG profile to use. One of the following based on gpu type\nPlease refer to https://docs.nvidia.com/datacenter/tesla/mig-user-guide/#supported-mig-profiles for more details\nA30 - [1g.6gb, 2g.12gb, 4g.24gb]\nA100 40 GB - [1g.5gb, 1g.10gb, 2g.10gb, 3g.20gb, 4g.20gb, 7g.40gb]\nA100 80 GB / H100 80 GB - [1g.10gb, 1g.20gb, 2g.20gb, 3g.40gb, 4g.40gb, 7g.80gb]\nH100 94 GB - [1g.12gb, 1g.24gb, 2g.24gb, 3g.47gb, 4g.47gb, 7g.94gb]\nH100 96 GB - [1g.12gb, 1g.24gb, 2g.24gb, 3g.48gb, 4g.48gb, 7g.96gb]\nH200 141 GB - [1g.18gb, 1g.35gb, 2g.35gb, 3g.71gb, 4g.71gb]\nB200 180 GB - [1g.23gb, 1g.45gb, 2g.45gb, 3g.90gb, 4g.90gb, 7g.180gb]",
     )
 
 
@@ -552,7 +567,7 @@ class NvidiaTimeslicingGPU(BaseModel):
         None,
         description="Name of the Nvidia GPU. One of [P4, P100, V100, T4, A10G, A100_40GB, A100_80GB]\nThis field is required for Node Selector and can be ignored in Nodepool Selector.\nOne instance of the card contains the following amount of memory -\nP4: 8 GB, P100: 16 GB, V100: 16 GB, T4: 16 GB, A10G: 24 GB, A100_40GB: 40GB, A100_80GB: 80 GB",
     )
-    gpu_memory: conint(ge=1, le=200000) = Field(
+    gpu_memory: conint(ge=1, le=400000) = Field(
         ...,
         description="Amount of GPU memory (in MB) to allocate. Please note, this limit is not being enforced today but will be in future. Applications are expected to operate in co-opertative mode",
     )
@@ -681,11 +696,11 @@ class Resources(BaseModel):
         0.5,
         description="CPU limit beyond which the usage cannot be exceeded. 1 CPU means 1 CPU core. Fractional CPU can be requested\nlike `0.5`. CPU limit should be >= cpu request.",
     )
-    memory_request: conint(ge=1, le=2000000) = Field(
+    memory_request: conint(ge=1, le=4000000) = Field(
         200,
         description="Requested memory which determines the minimum cost incurred. The unit of memory is in megabytes(MB).\nSo 1 means 1 MB and 2000 means 2GB.",
     )
-    memory_limit: conint(ge=1, le=2000000) = Field(
+    memory_limit: conint(ge=1, le=4000000) = Field(
         500,
         description="Memory limit after which the application will be killed with an OOM error. The unit of memory is\nin megabytes(MB). So 1 means 1 MB and 2000 means 2GB. MemoryLimit should be greater than memory request.",
     )
@@ -697,7 +712,7 @@ class Resources(BaseModel):
         2000,
         description="Disk storage limit. The unit of memory is in megabytes(MB). Exceeding this limit will result in eviction.\nIt should be greater than the request. This is ephemeral storage and will be wiped out on pod restarts or eviction",
     )
-    shared_memory_size: Optional[conint(ge=64, le=2000000)] = Field(
+    shared_memory_size: Optional[conint(ge=64, le=4000000)] = Field(
         None,
         description="Define the shared memory requirements for your workload. Machine learning libraries like Pytorch can use Shared Memory\nfor inter-process communication. If you use this, we will mount a `tmpfs` backed volume at the `/dev/shm` directory.\nAny usage will also count against the workload's memory limit (`resources.memory_limit`) along with your workload's memory usage.\nIf the overall usage goes above `resources.memory_limit` the user process may get killed.\nShared Memory Size cannot be more than the defined Memory Limit for the workload.",
     )
@@ -1236,7 +1251,7 @@ class ContainerTaskConfig(BaseModel):
         description="Configure environment variables to be injected in the task either as plain text or secrets. [Docs](https://docs.truefoundry.com/docs/env-variables)",
     )
     resources: Optional[Resources] = None
-    mounts: Optional[List[VolumeMount]] = Field(
+    mounts: Optional[List[Union[SecretMount, StringDataMount, VolumeMount]]] = Field(
         None, description="Configure data to be mounted to Workflow pod(s) as a volume."
     )
     service_account: Optional[str] = Field(None, description="")
@@ -1636,7 +1651,7 @@ class PySparkTaskConfig(BaseModel):
         None,
         description="Configure environment variables to be injected in the task either as plain text or secrets. [Docs](https://docs.truefoundry.com/docs/env-variables)",
     )
-    mounts: Optional[List[Union[SecretMount, StringDataMount, VolumeMount]]] = Field(
+    mounts: Optional[List[VolumeMount]] = Field(
         None, description="Configure data to be mounted to Workflow pod(s) as a volume."
     )
     service_account: Optional[str] = Field(None, description="")

truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py

@@ -80,7 +80,9 @@ CUDA_VERSION_TO_IMAGE_TAG: Dict[str, str] = {
     # From 12.4+ onwards, the image tags drop the cudnn version
     CUDAVersion.CUDA_12_4_CUDNN9.value: "12.4.1-cudnn-runtime-ubuntu22.04",
     CUDAVersion.CUDA_12_5_CUDNN9.value: "12.5.1-cudnn-runtime-ubuntu22.04",
-    CUDAVersion.CUDA_12_6_CUDNN9.value: "12.6.1-cudnn-runtime-ubuntu22.04",
+    CUDAVersion.CUDA_12_6_CUDNN9.value: "12.6.3-cudnn-runtime-ubuntu22.04",
+    CUDAVersion.CUDA_12_8_CUDNN9.value: "12.8.1-cudnn-runtime-ubuntu22.04",
+    CUDAVersion.CUDA_12_9_CUDNN9.value: "12.9.1-cudnn-runtime-ubuntu22.04",
 }
 
 

truefoundry/deploy/cli/commands/patch_command.py

@@ -58,9 +58,19 @@ def patch_command(
     ]
     if output_file:
         with open(output_file, "w") as fd:
-            subprocess.run(yq_command, stdout=fd)
+            p = subprocess.run(
+                yq_command, stdout=fd, stderr=subprocess.PIPE, check=False
+            )
+        if p.returncode != 0:
+            stderr = p.stderr.decode("UTF-8")
+            raise Exception(f"Failed to patch yaml file: {stderr}")
     else:
-        p = subprocess.run(yq_command, stdout=subprocess.PIPE)
+        p = subprocess.run(
+            yq_command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=False
+        )
+        if p.returncode != 0:
+            stderr = p.stderr.decode("UTF-8")
+            raise Exception(f"Failed to patch yaml file: {stderr}")
         output = p.stdout.decode("UTF-8")
         callback = RichOutputCallBack()
         callback.print_line(output)

truefoundry/deploy/v2/lib/patched_models.py

@@ -91,6 +91,8 @@ class CUDAVersion(str, enum.Enum):
     CUDA_12_4_CUDNN9 = "12.4-cudnn9"
     CUDA_12_5_CUDNN9 = "12.5-cudnn9"
     CUDA_12_6_CUDNN9 = "12.6-cudnn9"
+    CUDA_12_8_CUDNN9 = "12.8-cudnn9"
+    CUDA_12_9_CUDNN9 = "12.9-cudnn9"
 
 
 class GPUType(str, enum.Enum):
@@ -99,12 +101,19 @@ class GPUType(str, enum.Enum):
     V100 = "V100"
     T4 = "T4"
     A10G = "A10G"
+    A10_4GB = "A10_4GB"
+    A10_8GB = "A10_8GB"
+    A10_12GB = "A10_12GB"
+    A10_24GB = "A10_24GB"
     A100_40GB = "A100_40GB"
     A100_80GB = "A100_80GB"
     L4 = "L4"
+    L40S = "L40S"
     H100_80GB = "H100_80GB"
     H100_94GB = "H100_94GB"
+    H100_96GB = "H100_96GB"
     H200 = "H200"
+    B200 = "B200"
 
 
 class TPUType(str, enum.Enum):

{truefoundry-0.11.9.dist-info → truefoundry-0.11.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truefoundry
-Version: 0.11.9
+Version: 0.11.10
 Summary: TrueFoundry CLI
 Author-email: TrueFoundry Team <abhishek@truefoundry.com>
 Requires-Python: <3.14,>=3.8.1

{truefoundry-0.11.9.dist-info → truefoundry-0.11.10.dist-info}/RECORD

@@ -54,7 +54,7 @@ truefoundry/common/utils.py,sha256=P0FuAadoJGdpieUORLSN-PiFnkyoGO-K2cS4OPITBWg,6
 truefoundry/common/warnings.py,sha256=xDMhR_-ZGC40Ycaj6nlFb5MYPexn8WbKCHd4FlflTXQ,705
 truefoundry/deploy/__init__.py,sha256=sP-6Nv-_uV2o3knWcNSGV07j_Hkq0lfUkfZffBg-Hfo,2874
 truefoundry/deploy/python_deploy_codegen.py,sha256=WwP6bIzFoLpF7J2Bgef2HMSIeefJ8TWtSv4hXNycEzQ,8872
-truefoundry/deploy/_autogen/models.py,sha256=e75fSAlUJhPW3IN9Lg3ogSnCR9crIuHAsZaDSCNvkS0,75977
+truefoundry/deploy/_autogen/models.py,sha256=d1FjNfni3L3Az8d0zjT51N-TWSrW22GFC2wq8hFVoRo,76655
 truefoundry/deploy/builder/__init__.py,sha256=VR07ZB7ziONEBbVgg1JdRTWY7t4qJjJTMhc2VodXYdA,5036
 truefoundry/deploy/builder/constants.py,sha256=amUkHoHvVKzGv0v_knfiioRuKiJM0V0xW0diERgWiI0,508
 truefoundry/deploy/builder/docker_service.py,sha256=sm7GWeIqyrKaZpxskdLejZlsxcZnM3BTDJr6orvPN4E,3948
@@ -64,7 +64,7 @@ truefoundry/deploy/builder/builders/dockerfile.py,sha256=XMbMlPUTMPCyaHl7jJQY1OD
 truefoundry/deploy/builder/builders/tfy_notebook_buildpack/__init__.py,sha256=RGWGqY8xOF7vycUPJd10N7ZzahWv24lO0anrOPtLuDU,1796
 truefoundry/deploy/builder/builders/tfy_notebook_buildpack/dockerfile_template.py,sha256=rQgdvKmAT9HArVW4TAG5yd2QTKRs3S5LJ9RQbc_EkHE,2518
 truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py,sha256=_fjqHKn80qKi68SAMMALge7_A6e1sTsQWichw8uoGIw,2025
-truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py,sha256=Kj-ICGFTpDj86v6Juohz7q2TNYpcGIeKBW5HADG7SGE,6704
+truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py,sha256=bpYW8DN-fgRw85Vo3kSpigbjEkX4lELoI_iBXHmg_4Q,6856
 truefoundry/deploy/builder/builders/tfy_spark_buildpack/__init__.py,sha256=NEPlM6_vTVxp4ITa18B8DBbgYCn1q5d8be21lbgu5oY,2888
 truefoundry/deploy/builder/builders/tfy_spark_buildpack/dockerfile_template.py,sha256=nMJJfxjy8R7BZK89KicerQQwKLspUSJ3kerWZI3hFxk,4571
 truefoundry/deploy/builder/builders/tfy_spark_buildpack/tfy_execute_notebook.py,sha256=-D37Zjy2SBt3RHxonPEpR1_LR0W7vTSM1kQ1S-fdK-I,6363
@@ -84,7 +84,7 @@ truefoundry/deploy/cli/commands/login_command.py,sha256=GdFrH2zgFFrSZi35p6MPDkQi
 truefoundry/deploy/cli/commands/logout_command.py,sha256=u3kfrEp0ETbrz40KjD4GCC3XEZ5YRAlrca_Df4U_mk0,536
 truefoundry/deploy/cli/commands/logs_command.py,sha256=osl2z5VaIceB9sYa6GtwsuyAPZKcw-A0oVEt3g1f62Q,4140
 truefoundry/deploy/cli/commands/patch_application_command.py,sha256=aRTHu2OmxQd7j9iE0RavsFCkCILp0rGh4DJO51Oij5I,2591
-truefoundry/deploy/cli/commands/patch_command.py,sha256=wA95khMO9uVz8SaJlgYMUwaX7HagtchjyxXXATq83Bk,1665
+truefoundry/deploy/cli/commands/patch_command.py,sha256=F4M6aWzx1e7TKkDjvYr9S7A-S0vqkdVlMj0N_KO4_TI,2083
 truefoundry/deploy/cli/commands/terminate_comand.py,sha256=UKhOdbAej8ubX3q44vpLrOotAcvH4vHpRZJQrRf_AfM,1077
 truefoundry/deploy/cli/commands/trigger_command.py,sha256=-FZy_XnFsexH5SOWzDc4Dj9fTwmdjene_EaLjJPmG2c,5119
 truefoundry/deploy/cli/commands/utils.py,sha256=mIMYbHuAxnT0yz_0PU8LDC9sAZPU_xURZFMOrGoasuc,3694
@@ -119,7 +119,7 @@ truefoundry/deploy/v2/lib/deploy.py,sha256=Ltm7cpIW14IbmEsR3EAIeWQUch2Z6HLej7heu
 truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=G5BzMIbap8pgDX1eY-TITruUxQdkKhYtBmRwLL6lDeY,14342
 truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=mUi-OjPf7bc8rzfrPLdFb79LKuDq7F36RxL4V-AXebs,6830
 truefoundry/deploy/v2/lib/models.py,sha256=ogc1UYs1Z2nBdGSKCrde9sk8d0GxFKMkem99uqO5CmM,1148
-truefoundry/deploy/v2/lib/patched_models.py,sha256=bsznDLcUH5GcW8SUEvHETJqoFGlYYJ0j-tyGIqnRraw,16911
+truefoundry/deploy/v2/lib/patched_models.py,sha256=IFTIdAjXSG93f6ZgGC3gC2lXAsWq5p-ZqJGCmQ2Br5o,17149
 truefoundry/deploy/v2/lib/source.py,sha256=d6-8_6Zn5koBglqrBrY6ZLG_7yyPuLdyEmK4iZTw6xY,9405
 truefoundry/ml/__init__.py,sha256=EEEHV7w58Krpo_W9Chd8Y3TdItfFO3LI6j6Izqc4-P8,2219
 truefoundry/ml/constants.py,sha256=vDq72d4C9FSWqr9MMdjgTF4TuyNFApvo_6RVsSeAjB4,2837
@@ -387,7 +387,7 @@ truefoundry/workflow/remote_filesystem/__init__.py,sha256=LQ95ViEjJ7Ts4JcCGOxMPs
 truefoundry/workflow/remote_filesystem/logger.py,sha256=em2l7D6sw7xTLDP0kQSLpgfRRCLpN14Qw85TN7ujQcE,1022
 truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py,sha256=xcT0wQmQlgzcj0nP3tJopyFSVWT1uv3nhiTIuwfXYeg,12342
 truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py,sha256=nSGPZu0Gyd_jz0KsEE-7w_BmnTD8CVF1S8cUJoxaCbc,13305
-truefoundry-0.11.9.dist-info/METADATA,sha256=AxFp30YG9SHDmA5bBnahrDmMrLuykYApPy9EN5Bgs1Y,2760
-truefoundry-0.11.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truefoundry-0.11.9.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
-truefoundry-0.11.9.dist-info/RECORD,,
+truefoundry-0.11.10.dist-info/METADATA,sha256=ipgkIpx7Ho9NogRqOTZcBiOCnAMUE-BbOBHhvxE7i50,2761
+truefoundry-0.11.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truefoundry-0.11.10.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
+truefoundry-0.11.10.dist-info/RECORD,,