truefoundry 0.4.2rc1__py3-none-any.whl → 0.4.2rc2__py3-none-any.whl

truefoundry/deploy/auto_gen/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2024-09-29T10:28:22+00:00
+#   timestamp: 2024-10-09T13:00:44+00:00
 
 from __future__ import annotations
 
@@ -363,10 +363,12 @@ class HelmRepo(BaseModel):
         description="+label=Helm repository URL\n+sort=1\n+message=Needs to be a valid URL.",
     )
     chart: str = Field(
-        ..., description="+label=Chart name\n+sort=2\n+usage=The helm chart name"
+        ...,
+        description='+label=Chart name\n+sort=2\n+usage=The helm chart name\n+uiType=InputSelect\n+uiProps={"creatable":true, "searchable":true}',
     )
     version: str = Field(
-        ..., description="+label=Version\n+sort=3\n+usage=Helm chart version"
+        ...,
+        description='+label=Version\n+sort=3\n+usage=Helm chart version\n+uiType=InputSelect\n+uiProps={"creatable":true, "searchable":true}',
     )
 
 
@@ -448,10 +450,14 @@ class JobAlert(BaseModel):
     +label=Alert
     """
 
-    notification_channel: str = Field(
+    notification_channel: constr(min_length=1) = Field(
         ...,
         description='+label=Notification Channel\n+usage=Specify the notification channel to send alerts to\n+uiType=IntegrationSelect\n+uiProps={"integrationType":"notification-channel"}\n+sort=660',
     )
+    to_emails: Optional[List[constr(min_length=1)]] = Field(
+        None,
+        description="+label=To Emails\n+usage=List of recipients' email addresses if the notification channel is Email.\n+docs=Specify the emails to send alerts to\n+sort=665",
+    )
     on_start: bool = Field(
         False,
         description="+label=On Start\n+usage=Send an alert when the job starts\n+sort=670",
@@ -1546,7 +1552,7 @@ class SSHServer(BaseWorkbenchInput):
     image: WorkbenchImage
     ssh_public_key: str = Field(
         ...,
-        description="+label: SSH Public Key\n+usage=Add Your SSH Public Key, this will be used to authenticate you to the SSH Server. You can find it using `cat ~/.ssh/id_rsa.pub`\n+sort=4",
+        description="+label: SSH Public Key\n+usage=Add Your SSH Public Key, this will be used to authenticate you to the SSH Server.  \\\nYou can find it using `cat ~/.ssh/id_rsa.pub` in Mac/Linux or `type $home\\.ssh\\id_rsa.pub` in Windows Powershell.  \\\nYou can also generate a new SSH key pair using `ssh-keygen -t rsa` in your local terminal. (same for both Mac/Linux and Windows Powershell)\n+uiType=TextArea\n+sort=4",
     )
 
 

truefoundry/ml/mlfoundry_run.py

@@ -52,6 +52,8 @@ from truefoundry.ml.run_utils import ParamsType, flatten_dict, process_params
 from truefoundry.ml.session import ACTIVE_RUNS, _get_api_client, get_active_session
 from truefoundry.ml.validation_utils import (
     MAX_ENTITY_KEY_LENGTH,
+    MAX_METRICS_PER_BATCH,
+    MAX_PARAMS_TAGS_PER_BATCH,
     _validate_batch_log_data,
 )
 
@@ -638,12 +640,16 @@ class MlFoundryRun:
             return
 
         try:
-            _validate_batch_log_data(metrics=metrics, params=[], tags=[])
-            self._runs_api.log_run_batch_post(
-                log_batch_request_dto=LogBatchRequestDto(
-                    run_id=self.run_id, metrics=metrics, params=[], tags=[]
+            for i in range(0, len(metrics), MAX_METRICS_PER_BATCH):
+                metrics_batch = metrics[i : i + MAX_METRICS_PER_BATCH]
+
+                _validate_batch_log_data(metrics=metrics_batch, params=[], tags=[])
+
+                self._runs_api.log_run_batch_post(
+                    log_batch_request_dto=LogBatchRequestDto(
+                        run_id=self.run_id, metrics=metrics_batch, params=[], tags=[]
+                    )
                 )
-            )
         except Exception as e:
             raise MlFoundryException(str(e)) from e
 
@@ -725,12 +731,17 @@ class MlFoundryRun:
             if len(params) == 0:
                 logger.warning("Cannot log empty params dictionary")
 
-            _validate_batch_log_data(metrics=[], params=params, tags=[])
-            self._runs_api.log_run_batch_post(
-                log_batch_request_dto=LogBatchRequestDto(
-                    run_id=self.run_id, metrics=[], params=params, tags=[]
+            for i in range(0, len(params), MAX_PARAMS_TAGS_PER_BATCH):
+                params_batch = params[i : i + MAX_PARAMS_TAGS_PER_BATCH]
+
+                _validate_batch_log_data(metrics=[], params=params_batch, tags=[])
+                logger.debug("Logging parameters: %s", params_batch)
+
+                self._runs_api.log_run_batch_post(
+                    log_batch_request_dto=LogBatchRequestDto(
+                        run_id=self.run_id, metrics=[], params=params_batch, tags=[]
+                    )
                 )
-            )
         except Exception as e:
             raise MlFoundryException(str(e)) from e
         logger.info("Parameters logged successfully")
@@ -767,11 +778,15 @@ class MlFoundryRun:
             tags_arr = [
                 RunTagDto(key=key, value=str(value)) for key, value in tags.items()
             ]
-            self._runs_api.log_run_batch_post(
-                log_batch_request_dto=LogBatchRequestDto(
-                    run_id=self.run_id, metrics=[], params=[], tags=tags_arr
+            for i in range(0, len(tags_arr), MAX_PARAMS_TAGS_PER_BATCH):
+                tags_batch = tags_arr[i : i + MAX_PARAMS_TAGS_PER_BATCH]
+
+                _validate_batch_log_data(metrics=[], params=[], tags=tags_batch)
+                self._runs_api.log_run_batch_post(
+                    log_batch_request_dto=LogBatchRequestDto(
+                        run_id=self.run_id, metrics=[], params=[], tags=tags_batch
+                    )
                 )
-            )
         except Exception as e:
             raise MlFoundryException(str(e)) from e
         logger.info("Tags set successfully")

{truefoundry-0.4.2rc1.dist-info → truefoundry-0.4.2rc2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.4.2rc1
+Version: 0.4.2rc2
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com

{truefoundry-0.4.2rc1.dist-info → truefoundry-0.4.2rc2.dist-info}/RECORD

@@ -36,7 +36,7 @@ truefoundry/common/request_utils.py,sha256=-ss8033PClJOMJuS5Ue1zCFRVK7ZW9vjtva1b
 truefoundry/common/servicefoundry_client.py,sha256=qYSrsbFH8b7N_2oxlXfbJnWfdqHCBQ1dQ5Rr2tdp3zQ,2937
 truefoundry/common/utils.py,sha256=f5Av8X70vZf6eUK4Axm8q75It2-C-3o3uomA0GrFt_4,2506
 truefoundry/deploy/__init__.py,sha256=ugawKF2G02EmEXX35oZ2tec12d9oWN28Sf6mtGGIERY,2281
-truefoundry/deploy/auto_gen/models.py,sha256=I4-tV2V74mljN7lH606pCD5LArK6MvND_i7Vt_aRI5Q,79952
+truefoundry/deploy/auto_gen/models.py,sha256=AjbjNFwWDTIOjN_E1fSUgP-ej9nsb2OpZPhsnIRAVGs,80621
 truefoundry/deploy/builder/__init__.py,sha256=a1qR6nicHGcxRaeNTxWRsmDs8zsmXc7j13-I8q0qqVk,4938
 truefoundry/deploy/builder/builders/__init__.py,sha256=tlFLXqyDaKLd4iZbo4Hcu_8gOmgtL6drnXpbmQ6x1P8,636
 truefoundry/deploy/builder/builders/dockerfile.py,sha256=AXXTziCkaqIhuM_bwyD1vT1znOwemN1TKgU7eyo-KuM,1522
@@ -323,7 +323,7 @@ truefoundry/ml/log_types/pydantic_base.py,sha256=eBlw_AEyAz4iJKDP4zgJOCFWcldwQqp
 truefoundry/ml/log_types/utils.py,sha256=xjJ21jdPScvFmw3TbVh5NCzbzJwaqiXJyiiT4xxX1EI,335
 truefoundry/ml/logger.py,sha256=VT-BF3BnBYTWVq87O58F0c8uXMu94gYzsiFlGY3_7Ao,458
 truefoundry/ml/mlfoundry_api.py,sha256=I2T8tXeAIWpD8EH05fm80mNyX6cs2S1ORI4qoo0HTpQ,60847
-truefoundry/ml/mlfoundry_run.py,sha256=gl1aA_j7KwtyOpZgTTi3boWxpdTyssGGFh_lP40mc1o,44387
+truefoundry/ml/mlfoundry_run.py,sha256=rNJl130iJkpjW3MNoe5-d_J9VJJQBqWHEJCfYeiZCbE,45123
 truefoundry/ml/run_utils.py,sha256=dzbQ_TGkSCYO-gg8tlr5tglR0p2SIQdy0wl4IEQE5JA,2899
 truefoundry/ml/session.py,sha256=_WmaGNiZCwQVmB-brbX-z38nFvuqP8f7jGxfqJoq3TM,5385
 truefoundry/ml/validation_utils.py,sha256=XBSUd9OoyriWJpT3M5LKz17iWY3yVMr3hM5vdaVjtf0,12082
@@ -340,7 +340,7 @@ truefoundry/workflow/map_task.py,sha256=2m3qGXQ90k9LdS45q8dqCCECc3qr8t2m_LMCVd1m
 truefoundry/workflow/python_task.py,sha256=SRXRLC4vdBqGjhkwuaY39LEWN6iPCpJAuW17URRdWTY,1128
 truefoundry/workflow/task.py,sha256=ToitYiKcNzFCtOVQwz1W8sRjbR97eVS7vQBdbgUQtKg,1779
 truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.4.2rc1.dist-info/METADATA,sha256=WmKC8UIgG2oVxwmlurcKI9T3aL0TTxfB8oiKuywh-uU,3140
-truefoundry-0.4.2rc1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-truefoundry-0.4.2rc1.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.4.2rc1.dist-info/RECORD,,
+truefoundry-0.4.2rc2.dist-info/METADATA,sha256=8wBExyVvKXs8p3xYsQf8p6uGf8Mb16m8DK2GlgHt78o,3140
+truefoundry-0.4.2rc2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+truefoundry-0.4.2rc2.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.4.2rc2.dist-info/RECORD,,