truefoundry 0.11.5rc2__py3-none-any.whl → 0.11.7__py3-none-any.whl

truefoundry/deploy/python_deploy_codegen.py

@@ -187,7 +187,10 @@ def convert_deployment_config_to_python(
 
 
 def generate_python_snippet_for_trigger_job(
-    application_fqn: str, command: Optional[str], params: Optional[Dict[str, str]]
+    application_fqn: str,
+    command: Optional[str],
+    params: Optional[Dict[str, str]],
+    run_name_alias: Optional[str],
 ):
     job_run_python_template = """\
 from truefoundry.deploy import trigger_job
@@ -195,8 +198,9 @@ from truefoundry.deploy import trigger_job
 response = trigger_job(
   application_fqn="{{application_fqn}}",
   # You can pass command or params, but not both
-  # command={{command}}
-  # params={{params}}
+  # command={{command}},
+  # params={{params}},
+  # run_name_alias={{run_name_alias}}
 )
 
 print(response.jobRunName)
@@ -220,6 +224,17 @@ print(response.jobRunName)
         output_python_str = output_python_str.replace(
             "{{params}}", "<Enter Params(key-value pairs) here as python dict>"
         )
+    if run_name_alias is not None:
+        output_python_str = output_python_str.replace(
+            "# run_name_alias", "run_name_alias"
+        )
+        output_python_str = output_python_str.replace(
+            "{{run_name_alias}}", repr(run_name_alias)
+        )
+    else:
+        output_python_str = output_python_str.replace(
+            "{{run_name_alias}}", "<Enter Run Name Alias here>"
+        )
 
     return output_python_str
 
@@ -229,6 +244,7 @@ def generate_curl_snippet_for_trigger_job(
     application_id: str,
     command: Optional[str],
     params: Optional[Dict[str, str]],
+    run_name_alias: Optional[str],
 ):
     job_run_curl_request_template = """curl -X 'POST' \\
   '{{control_plane_url}}/api/svc/v1/jobs/trigger' \\
@@ -237,7 +253,10 @@ def generate_curl_snippet_for_trigger_job(
   -H 'Content-Type: application/json' \\
   -d '{
   "applicationId": "{{application_id}}",
-  "input": {{input}}
+  "input": {{input}},
+  "metadata": {
+    "job_run_name_alias": "{{run_name_alias}}"
+  }
 }'
 """
     output_curl_str = job_run_curl_request_template.replace(
@@ -247,4 +266,10 @@ def generate_curl_snippet_for_trigger_job(
     output_curl_str = output_curl_str.replace(
         "{{input}}", json.dumps({"command": command, "params": params}, indent=2)
     )
+    if run_name_alias is not None:
+        output_curl_str = output_curl_str.replace("{{run_name_alias}}", run_name_alias)
+    else:
+        output_curl_str = output_curl_str.replace(
+            "{{run_name_alias}}", "<Enter Run Name Alias here>"
+        )
     return output_curl_str

truefoundry/ml/integrations/huggingface/trainer_callback.py

@@ -31,6 +31,7 @@ class TrueFoundryMLCallback(TrainerCallback):
         log_checkpoints: bool = True,
         checkpoint_artifact_name: Optional[str] = None,
         auto_end_run_on_train_end: bool = False,
+        log_training_arguments: bool = True,
     ):
         """
         Args:
@@ -38,7 +39,7 @@ class TrueFoundryMLCallback(TrainerCallback):
             log_checkpoints: Whether to log checkpoints or not, defaults to True.
             checkpoint_artifact_name: The name of the artifact to log checkpoints to, required if log_checkpoints is True.
             auto_end_run_on_train_end: Whether to end the run automatically when training ends, defaults to False.
-
+            log_training_arguments: Whether to log the training arguments or not, defaults to True.
             Usage:
                 from transformers import Trainer
                 from truefoundry.ml.integrations.huggingface.trainer_callback import TrueFoundryMLCallback
@@ -67,6 +68,7 @@ class TrueFoundryMLCallback(TrainerCallback):
             )
         self._checkpoint_artifact_name = checkpoint_artifact_name
         self._auto_end_run_on_train_end = auto_end_run_on_train_end
+        self._log_training_arguments = log_training_arguments
 
     @classmethod
     def with_managed_run(
@@ -76,6 +78,7 @@ class TrueFoundryMLCallback(TrainerCallback):
         log_checkpoints: bool = True,
         checkpoint_artifact_name: Optional[str] = None,
         auto_end_run_on_train_end: bool = True,
+        log_training_arguments: bool = True,
     ) -> "TrueFoundryMLCallback":
         """
         Args:
@@ -84,7 +87,7 @@ class TrueFoundryMLCallback(TrainerCallback):
             log_checkpoints: Whether to log checkpoints or not, defaults to True.
             checkpoint_artifact_name: The name of the artifact to log checkpoints to, required if log_checkpoints is True.
             auto_end_run_on_train_end: Whether to end the run automatically when training ends, defaults to True.
-
+            log_training_arguments: Whether to log the training arguments or not, defaults to True.
         Usage:
             from transformers import Trainer
             from truefoundry.ml.integrations.huggingface.trainer_callback import TrueFoundryMLCallback
@@ -109,6 +112,7 @@ class TrueFoundryMLCallback(TrainerCallback):
             log_checkpoints=log_checkpoints,
             checkpoint_artifact_name=checkpoint_artifact_name,
             auto_end_run_on_train_end=auto_end_run_on_train_end,
+            log_training_arguments=log_training_arguments,
         )
 
     def _drop_non_finite_values(self, dct: Dict[str, Any]) -> Dict[str, Any]:
@@ -184,6 +188,21 @@ class TrueFoundryMLCallback(TrainerCallback):
             description=description,
         )
 
+    def on_train_begin(
+        self,
+        args: "TrainingArguments",
+        state: "TrainerState",
+        control: "TrainerControl",
+        **kwargs,
+    ):
+        if self._log_training_arguments:
+            training_arguments = args.to_sanitized_dict()
+            valid_types = (bool, int, float, str)
+            training_arguments = {
+                k: v for k, v in training_arguments.items() if type(v) in valid_types
+            }
+            self._run.log_params(training_arguments)
+
     def on_train_end(
         self,
         args: "TrainingArguments",

{truefoundry-0.11.5rc2.dist-info → truefoundry-0.11.7.dist-info}/METADATA

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

{truefoundry-0.11.5rc2.dist-info → truefoundry-0.11.7.dist-info}/RECORD

@@ -53,7 +53,7 @@ truefoundry/common/types.py,sha256=BMJFCsR1lPJAw66IQBSvLyV4I6o_x5oj78gVsUa9si8,1
 truefoundry/common/utils.py,sha256=P0FuAadoJGdpieUORLSN-PiFnkyoGO-K2cS4OPITBWg,6714
 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=k19_m5DGsUyjOUCSKwIVP8vDna2sq01tHABsUfoVpW4,8019
+truefoundry/deploy/python_deploy_codegen.py,sha256=WwP6bIzFoLpF7J2Bgef2HMSIeefJ8TWtSv4hXNycEzQ,8872
 truefoundry/deploy/_autogen/models.py,sha256=e75fSAlUJhPW3IN9Lg3ogSnCR9crIuHAsZaDSCNvkS0,75977
 truefoundry/deploy/builder/__init__.py,sha256=VR07ZB7ziONEBbVgg1JdRTWY7t4qJjJTMhc2VodXYdA,5036
 truefoundry/deploy/builder/constants.py,sha256=amUkHoHvVKzGv0v_knfiioRuKiJM0V0xW0diERgWiI0,508
@@ -359,7 +359,7 @@ truefoundry/ml/cli/commands/model_init.py,sha256=INyUAU6hiFClI8cZqX5hgnrtNbeKxlZ
 truefoundry/ml/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/integrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/integrations/huggingface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/ml/integrations/huggingface/trainer_callback.py,sha256=Zu5AUbH_ct8I1dHyNYJQZBj9Y__hKo0sc2OxpPXJARE,6952
+truefoundry/ml/integrations/huggingface/trainer_callback.py,sha256=Q_CT1Sy4OZ8LyHm2qruhAmp3H7emhJogSs2-hPsPK3k,7863
 truefoundry/ml/log_types/__init__.py,sha256=g4u4D4Jaj0aBK5GtrLV88-qThKZR9pSZ17vFEkN-LmM,125
 truefoundry/ml/log_types/plot.py,sha256=LDh4uy6z2P_a2oPM2lc85c0lt8utVvunohzeMawFjZw,7572
 truefoundry/ml/log_types/pydantic_base.py,sha256=eBlw_AEyAz4iJKDP4zgJOCFWcldwQqpf7FADW1jzIQY,272
@@ -386,7 +386,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.5rc2.dist-info/METADATA,sha256=Pxl2_AUdbaHgeqxSo2h-b-gFmtLRGV1Ew5c0-2FX6jE,2763
-truefoundry-0.11.5rc2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truefoundry-0.11.5rc2.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
-truefoundry-0.11.5rc2.dist-info/RECORD,,
+truefoundry-0.11.7.dist-info/METADATA,sha256=ydC2lFQf5iICc9uFTslsw4zS96saw6rWGDbcEup0zSc,2760
+truefoundry-0.11.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truefoundry-0.11.7.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
+truefoundry-0.11.7.dist-info/RECORD,,