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()
