def update_training_job_prefix_if_not_specified()

in smdebug_rulesconfig/actions/actions.py [0:0]

• 3 lines of code
• 2 McCabe index (conditional complexity)

    def update_training_job_prefix_if_not_specified(self, training_job_name: str):
        """
        Update the training job prefix to be the training job name if the user has not already specified a custom
        training job prefix. This is only meant to be called via the sagemaker SDK when `estimator.fit` is called by the
        user. Validation is purposely excluded here so that any failures in validation of the training job name are
        intentionally caught in the sagemaker SDK and not here.

        :param training_job_name: Name of the training job, passed in when `estimator.fit` is called.
        """
        if self.use_default_training_job_prefix:
            self.action_parameters["training_job_prefix"] = training_job_name