Skip to content

Commit fbb2cca

Browse files
committed
fix: add label_schema_id to job creation
1 parent f64802b commit fbb2cca

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

sdk/diffgram/file/view.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ def get_label_file_dict(self, schema_id = None, use_session = True):
4242

4343
endpoint = "/api/v1/project/" + self.project_string_id + \
4444
"/labels/view/name_to_file_id"
45-
params = {'schema_id': schema_id}
45+
params = {'schema_id': schema_id, 'directory_id': self.directory_id}
4646
if use_session:
4747
response = self.session.get(self.host + endpoint, params = params)
4848
else:

sdk/diffgram/job/job.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,7 @@ def serialize(self):
121121
'id': self.id,
122122
'name': self.name,
123123
'status': self.status,
124+
'label_schema_id': self.label_schema_id,
124125
'instance_type': self.instance_type,
125126
'share': self.share,
126127
'type': self.type,
@@ -142,6 +143,7 @@ def new(self,
142143
instance_type=None,
143144
share="project",
144145
job_type="Normal",
146+
label_schema_id = None,
145147
permission=None,
146148
field=None,
147149
category=None,
@@ -188,6 +190,7 @@ def new(self,
188190
job.passes_per_file = passes_per_file
189191
job.launch_datetime = launch_datetime
190192
job.label_file_list = label_file_list
193+
job.label_schema_id = label_schema_id
191194
job.attached_directories = []
192195
job.member_list_ids = members_list_ids
193196

0 commit comments

Comments
 (0)