def find_dataset_by(self, dataset_id): return DatasetModel.find_by(dataset_id)
def execute(self, dataset_id, dataset_source_folder_path): dataset = DatasetModel.find_by(dataset_id) self.move_content_from_source_to_dataset(dataset, dataset_source_folder_path) self.image_repository.create_images_from(dataset.id) return self.scan_annotation_from_json(dataset_id, dataset_source_folder_path)
def execute(self, dataset_id, categories): dataset = DatasetModel.find_by(dataset_id) category_id_list = CategoryModel.bulk_create(categories) # dataset.update(set__categories=category_id_list) dataset.update_categories(category_id_list)
def set_dataset_by(self, dataset_id): self.dataset = DatasetModel.find_by(dataset_id)