ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-01 09:36:49 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix TriggerScheduler on Dataset change (issue #491)

Browse Source
This commit is contained in:
allegroai 2021-11-26 16:39:32 +02:00
parent c3b11b06f1
commit 62a175a243
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
clearml/automation/trigger.py
View File

@ -89,11 +89,16 @@ class DatasetTrigger(BaseTrigger):
def build_query(self, ref_time): def build_query(self, ref_time):
query = super(DatasetTrigger, self).build_query(ref_time) query = super(DatasetTrigger, self).build_query(ref_time)
if self.on_publish: query.update({
query.update({'status': ['published']}) 'system_tags': list(set(query.get('system_tags', []) + ['dataset'])),
'task_types': list(set(query.get('task_types', []) + [str(Task.TaskTypes.data_processing)])),
'status': ['published' if self.on_publish else 'completed']
})
if self.on_archive: if self.on_archive:
system_tags = list(set(query.get('system_tags', []) + ['archived'])) system_tags = list(set(query.get('system_tags', []) + ['archived']))
query.update({'system_tags': system_tags}) query.update({'system_tags': system_tags})
return query return query