def upload_catalog(self, catalog: Catalog) -> None: """Upload a catalog to the draft. Arguments: catalog: :class:`~tensorbay.label.catalog.Catalog` to upload. """ self._status.check_authority_for_draft() put_data: Dict[str, Any] = {"catalog": catalog.dumps()} put_data.update(self._status.get_status_info()) self._client.open_api_do("PUT", "labels/catalogs", self._dataset_id, json=put_data)
def test_dumps(self, categories, keypoints, catalog_contents): catalog = Catalog() catalog.box2d = Box2DSubcatalog() catalog.box3d = Box3DSubcatalog() catalog.polygon = PolygonSubcatalog() catalog.polyline2d = Polyline2DSubcatalog() catalog.sentence = SentenceSubcatalog() classificationsubcatalog = ClassificationSubcatalog() classificationsubcatalog.categories = categories catalog.classification = classificationsubcatalog keypoints2dsubcatalog = Keypoints2DSubcatalog() keypoints2dsubcatalog._keypoints = [keypoints] catalog.keypoints2d = keypoints2dsubcatalog assert catalog.dumps() == catalog_contents