Skip to content

Commit 0f914fd

Browse files
committed
feat: add better support for export params
1 parent 21a10a1 commit 0f914fd

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

Diff for: sdk/diffgram/job/job.py

+6-3
Original file line numberDiff line numberDiff line change
@@ -388,6 +388,9 @@ def generate_export(
388388
self,
389389
kind='Annotations',
390390
return_type="data",
391+
source="job",
392+
masks=False,
393+
directory_id=None,
391394
wait_for_export_generation=True,
392395
ann_is_complete=None # Bool. None=='all', True=='complete' tasks only
393396
):
@@ -417,10 +420,10 @@ def generate_export(
417420
spec_dict = {
418421
'job_id': self.id,
419422
'kind': kind,
420-
'source': "job",
423+
'source': source,
421424
'file_comparison_mode': "latest",
422-
'directory_id': None,
423-
'masks': False,
425+
'directory_id': directory_id,
426+
'masks': masks,
424427
'ann_is_complete': ann_is_complete,
425428
'return_type': return_type,
426429
'wait_for_export_generation': wait_for_export_generation

0 commit comments

Comments
 (0)