From 80fede8b1dbedb63df3e48230329695620b4cfdf Mon Sep 17 00:00:00 2001 From: Carson Davis Date: Thu, 27 Mar 2025 10:38:35 -0500 Subject: [PATCH] specify exact collections on which to run TDAMM classifications --- sde_collections/models/collection.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/sde_collections/models/collection.py b/sde_collections/models/collection.py index 52a520d9..097c5ce7 100644 --- a/sde_collections/models/collection.py +++ b/sde_collections/models/collection.py @@ -691,12 +691,17 @@ def generate_inference_job(self, classification_type): def queue_necessary_classifications(self): """Check if collection needs classification and queue jobs if needed""" - - # Determine which classifications are needed - if self.division == Divisions.ASTROPHYSICS: + tdamm_collections = [ + "imagine_the_universe", + "physics_of_the_cosmos", + "stsci_space_telescope_science_institute", + ] + if self.config_folder in tdamm_collections: self.generate_inference_job(ClassificationType.TDAMM) - elif self.division == Divisions.GENERAL: - self.generate_inference_job(ClassificationType.DIVISION) + # if self.division == Divisions.ASTROPHYSICS: + # self.generate_inference_job(ClassificationType.TDAMM) + # elif self.division == Divisions.GENERAL: + # self.generate_inference_job(ClassificationType.DIVISION) else: # No classification needed, proceed directly to migration migrate_dump_to_delta_and_handle_status_transistions.delay(self.id)