diff --git a/sdk/diffgram/job/job.py b/sdk/diffgram/job/job.py index 4059399..df01702 100644 --- a/sdk/diffgram/job/job.py +++ b/sdk/diffgram/job/job.py @@ -373,7 +373,8 @@ def generate_export( self, kind='Annotations', return_type="data", - wait_for_export_generation=True + wait_for_export_generation=True, + ann_is_complete=None # Bool. None=='all', True=='complete' tasks only ): """ @@ -405,6 +406,7 @@ def generate_export( 'file_comparison_mode': "latest", 'directory_id': None, 'masks': False, + 'ann_is_complete': ann_is_complete, 'return_type': return_type, 'wait_for_export_generation': wait_for_export_generation }