From fbb2ccaa55d32a411ec635ba58d512e420d97643 Mon Sep 17 00:00:00 2001 From: Pablo Date: Fri, 3 Jun 2022 16:33:44 -0600 Subject: [PATCH] fix: add label_schema_id to job creation --- sdk/diffgram/file/view.py | 2 +- sdk/diffgram/job/job.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/sdk/diffgram/file/view.py b/sdk/diffgram/file/view.py index d4c8014..1cdb1c8 100644 --- a/sdk/diffgram/file/view.py +++ b/sdk/diffgram/file/view.py @@ -42,7 +42,7 @@ def get_label_file_dict(self, schema_id = None, use_session = True): endpoint = "/api/v1/project/" + self.project_string_id + \ "/labels/view/name_to_file_id" - params = {'schema_id': schema_id} + params = {'schema_id': schema_id, 'directory_id': self.directory_id} if use_session: response = self.session.get(self.host + endpoint, params = params) else: diff --git a/sdk/diffgram/job/job.py b/sdk/diffgram/job/job.py index a9f60da..80e5c6e 100644 --- a/sdk/diffgram/job/job.py +++ b/sdk/diffgram/job/job.py @@ -121,6 +121,7 @@ def serialize(self): 'id': self.id, 'name': self.name, 'status': self.status, + 'label_schema_id': self.label_schema_id, 'instance_type': self.instance_type, 'share': self.share, 'type': self.type, @@ -142,6 +143,7 @@ def new(self, instance_type=None, share="project", job_type="Normal", + label_schema_id = None, permission=None, field=None, category=None, @@ -188,6 +190,7 @@ def new(self, job.passes_per_file = passes_per_file job.launch_datetime = launch_datetime job.label_file_list = label_file_list + job.label_schema_id = label_schema_id job.attached_directories = [] job.member_list_ids = members_list_ids