def test_remove_classifier_data(self) -> None: """Test that classifier data is removed upon deletion.""" fs_services.save_classifier_data('exp_id', 'job_id', self.classifier_data_proto) self.assertTrue(self.fs.isfile('job_id-classifier-data.pb.xz')) fs_services.delete_classifier_data('exp_id', 'job_id') self.assertFalse(self.fs.isfile('job_id-classifier-data.pb.xz'))
def delete_classifier_training_job(job_id): """Deletes classifier training job model in the datastore given job_id. Args: job_id: str. ID of the classifier training job. """ classifier_training_job_model = ( classifier_models.ClassifierTrainingJobModel.get(job_id)) if classifier_training_job_model is not None: fs_services.delete_classifier_data( classifier_training_job_model.exp_id, job_id) classifier_training_job_model.delete()