rapidata 1.10.1__py3-none-any.whl → 2.1.0__py3-none-any.whl

rapidata/rapidata_client/dataset/validation_set_builder.py

@@ -1,336 +0,0 @@
-import os
-from rapidata.api_client.models.attach_category_truth import AttachCategoryTruth
-from rapidata.api_client.models.classify_payload import ClassifyPayload
-from rapidata.api_client.models.compare_payload import ComparePayload
-from rapidata.api_client.models.compare_truth import CompareTruth
-from rapidata.api_client.models.transcription_payload import TranscriptionPayload
-from rapidata.api_client.models.transcription_truth import TranscriptionTruth
-from rapidata.api_client.models.transcription_word import TranscriptionWord
-from rapidata.rapidata_client.assets.media_asset import MediaAsset
-from rapidata.rapidata_client.assets.multi_asset import MultiAsset
-from rapidata.rapidata_client.assets.text_asset import TextAsset
-from rapidata.rapidata_client.dataset.rapidata_validation_set import (
-    RapidataValidationSet,
-)
-from rapidata.rapidata_client.dataset.validation_rapid_parts import ValidatioRapidParts
-from rapidata.rapidata_client.metadata.base_metadata import Metadata
-from rapidata.service.openapi_service import OpenAPIService
-
-from rapidata.rapidata_client.dataset.rapid_builders.rapids import (
-    Rapid, 
-    ClassificationRapid,
-    CompareRapid,
-    SelectWordsRapid
-)
-from deprecated import deprecated
-
-
-class ValidationSetBuilder:
-    """The ValidationSetBuilder is used to build a validation set.
-    Give the validation set a name and then add classify, compare, or transcription rapid parts to it.
-    Get a `ValidationSetBuilder` by calling [`rapi.new_validation_set()`](../rapidata_client.md/#rapidata.rapidata_client.rapidata_client.RapidataClient.new_validation_set).
-    """
-
-    def __init__(self, name: str, openapi_service: OpenAPIService):
-        """
-        Initialize the ValidationSetBuilder.
-
-        Args:
-            name (str): The name of the validation set.
-            openapi_service (OpenAPIService): An instance of OpenAPIService to interact with the API.
-        """
-        self.name = name
-        self.openapi_service = openapi_service
-        self.validation_set_id: str | None = None
-        self._rapid_parts: list[ValidatioRapidParts] = []
-
-    def submit(self) -> RapidataValidationSet:
-        """Create the validation set by executing all HTTP requests. This should be the last method called on the builder.
-
-        Returns:
-            RapidataValidationSet: A RapidataValidationSet instance.
-
-        Raises:
-            ValueError: If the validation set creation fails.
-        """
-        result = (
-            self.openapi_service.validation_api.validation_create_validation_set_post(
-                name=self.name
-            )
-        )
-        self.validation_set_id = result.validation_set_id
-
-        if self.validation_set_id is None:
-            raise ValueError("Failed to create validation set")
-
-        validation_set = RapidataValidationSet(
-            validation_set_id=self.validation_set_id,
-            openapi_service=self.openapi_service,
-            name=self.name,
-        )
-
-        for rapid_part in self._rapid_parts:
-            validation_set.add_general_validation_rapid(
-                payload=rapid_part.payload,
-                truths=rapid_part.truths,
-                metadata=rapid_part.metadata,
-                asset=rapid_part.asset,
-                randomCorrectProbability=rapid_part.randomCorrectProbability,
-            )
-
-        return validation_set
-
-    @deprecated("Use submit instead")
-    def create(self) -> RapidataValidationSet:
-        """Create the validation set by executing all HTTP requests. This should be the last method called on the builder.
-
-        Returns:
-            RapidataValidationSet: A RapidataValidationSet instance.
-
-        Raises:
-            ValueError: If the validation set creation fails.
-        """
-        return self.submit()
-    
-    def add_rapid(self, rapid: Rapid):
-        """Add a rapid to the validation set.
-        To create the Rapid, use the RapidataClient.rapid_builder instance.
-
-        Args:
-            rapid (Rapid): The rapid to add to the validation set.
-        """
-        if not isinstance(rapid, Rapid):
-            raise ValueError("This method only accepts Rapid instances")
-        
-        if isinstance(rapid, ClassificationRapid):
-            self._add_classify_rapid(rapid.asset, rapid.question, rapid.options, rapid.truths, rapid.metadata)
-
-        if isinstance(rapid, CompareRapid):
-            self._add_compare_rapid(rapid.asset, rapid.criteria, rapid.truth, rapid.metadata)
-
-        if isinstance(rapid, SelectWordsRapid):
-            self._add_select_words_rapid(rapid.asset, rapid.instruction, rapid.text, rapid.truths, rapid.strict_grading)
-
-        return self
-
-    @deprecated("Use add_rapid instead")
-    def add_classify_rapid(
-        self,
-        asset: MediaAsset | TextAsset,
-        question: str,
-        categories: list[str],
-        truths: list[str],
-        metadata: list[Metadata] = [],
-    ):
-        """Add a classify rapid to the validation set.
-
-        Args:
-            asset (MediaAsset | TextAsset): The asset for the rapid.
-            question (str): The question for the rapid.
-            categories (list[str]): The list of categories for the rapid.
-            truths (list[str]): The list of truths for the rapid.
-            metadata (list[Metadata], optional): The metadata for the rapid. Defaults to an empty list.
-
-        Returns:
-            ValidationSetBuilder: The ValidationSetBuilder instance.
-
-        Raises:
-            ValueError: If the lengths of categories and truths are inconsistent.
-        """
-        self._add_classify_rapid(asset, question, categories, truths, metadata)
-
-        return self
-    
-    def _add_classify_rapid(
-        self,
-        asset: MediaAsset | TextAsset,
-        question: str,
-        categories: list[str],
-        truths: list[str],
-        metadata: list[Metadata] = [],
-    ):
-        """Add a classify rapid to the validation set.
-
-        Args:
-            asset (MediaAsset | TextAsset): The asset for the rapid.
-            question (str): The question for the rapid.
-            categories (list[str]): The list of categories for the rapid.
-            truths (list[str]): The list of truths for the rapid.
-            metadata (list[Metadata], optional): The metadata for the rapid. Defaults to an empty list.
-
-        Returns:
-            ValidationSetBuilder: The ValidationSetBuilder instance.
-
-        Raises:
-            ValueError: If the lengths of categories and truths are inconsistent.
-        """
-        payload = ClassifyPayload(
-            _t="ClassifyPayload", possibleCategories=categories, title=question
-        )
-        model_truth = AttachCategoryTruth(
-            correctCategories=truths, _t="AttachCategoryTruth"
-        )
-
-        self._rapid_parts.append(
-            ValidatioRapidParts(
-                question=question,
-                payload=payload,
-                truths=model_truth,
-                metadata=metadata,
-                randomCorrectProbability=len(truths) / len(categories),
-                asset=asset,
-            )
-        )
-
-    @deprecated("Use add_rapid instead")
-    def add_compare_rapid(
-        self,
-        asset: MultiAsset,
-        question: str,
-        truth: str,
-        metadata: list[Metadata] = [],
-    ):
-        """Add a compare rapid to the validation set.
-
-        Args:
-            asset (MultiAsset): The assets for the rapid.
-            question (str): The question for the rapid.
-            truth (str): The truth identifier for the rapid.
-            metadata (list[Metadata], optional): The metadata for the rapid. Defaults to an empty list.
-
-        Returns:
-            ValidationSetBuilder: The ValidationSetBuilder instance.
-
-        Raises:
-            ValueError: If the number of assets is not exactly two.
-        """
-        self._add_compare_rapid(asset, question, truth, metadata)
-
-        return self
-    
-    def _add_compare_rapid(
-        self,
-        asset: MultiAsset,
-        criteria: str,
-        truth: str,
-        metadata: list[Metadata] = [],
-    ):
-        """Add a compare rapid to the validation set.
-
-        Args:
-            asset (MultiAsset): The assets for the rapid.
-            criteria (str): The criteria for the comparison.
-            truth (str): The truth identifier for the rapid.
-            metadata (list[Metadata], optional): The metadata for the rapid. Defaults to an empty list.
-
-        Returns:
-            ValidationSetBuilder: The ValidationSetBuilder instance.
-
-        Raises:
-            ValueError: If the number of assets is not exactly two.
-        """
-        payload = ComparePayload(_t="ComparePayload", criteria=criteria)
-        # take only last part of truth path
-        truth = os.path.basename(truth)
-        model_truth = CompareTruth(_t="CompareTruth", winnerId=truth)
-
-        if len(asset) != 2:
-            raise ValueError("Compare rapid requires exactly two media paths")
-
-        self._rapid_parts.append(
-            ValidatioRapidParts(
-                question=criteria,
-                payload=payload,
-                truths=model_truth,
-                metadata=metadata,
-                randomCorrectProbability=1 / len(asset),
-                asset=asset,
-            )
-        )
-
-    @deprecated("Use add_rapid instead")
-    def add_select_words_rapid(
-        self,
-        asset: MediaAsset | TextAsset,
-        question: str,
-        select_words: str,
-        truths: list[int],
-        strict_grading: bool | None = None,
-        metadata: list[Metadata] = [],
-    ):
-        """Add a select words rapid to the validation set.
-
-        Args:
-            asset (MediaAsset | TextAsset): The asset for the rapid.
-            question (str): The question for the rapid.
-            select words (list[str]): The select words for the rapid.
-            truths (list[int]): The list of indices of the true word selections.
-            strict_grading (bool | None, optional): The strict grading for the rapid. Defaults to None.
-            metadata (list[Metadata], optional): The metadata for the rapid.
-
-        Returns:
-            ValidationSetBuilder: The ValidationSetBuilder instance.
-
-        Raises:
-            ValueError: If a correct word is not found in the select words.
-        """
-        self._add_select_words_rapid(asset, question, select_words, truths, strict_grading, metadata)
-
-        return self
-    
-    def _add_select_words_rapid(
-        self,
-        asset: MediaAsset | TextAsset,
-        question: str,
-        select_words: str,
-        truths: list[int],
-        strict_grading: bool | None = None,
-        metadata: list[Metadata] = [],
-    ):
-        """Add a select words rapid to the validation set.
-
-        Args:
-            asset (MediaAsset | TextAsset): The asset for the rapid.
-            question (str): The question for the rapid.
-            select words (list[str]): The select words for the rapid.
-            truths (list[int]): The list of indices of the true word selections.
-            strict_grading (bool | None, optional): The strict grading for the rapid. Defaults to None.
-            metadata (list[Metadata], optional): The metadata for the rapid.
-
-        Returns:
-            ValidationSetBuilder: The ValidationSetBuilder instance.
-
-        Raises:
-            ValueError: If a correct word is not found in the select words.
-        """
-        transcription_words = [
-            TranscriptionWord(word=word, wordIndex=i)
-            for i, word in enumerate(select_words.split())
-        ]
-
-        true_words = []
-        for idx in truths:
-            if idx > len(transcription_words) - 1:
-                raise ValueError(f"Index {idx} is out of bounds")
-            true_words.append(transcription_words[idx])
-
-        payload = TranscriptionPayload(
-            _t="TranscriptionPayload", title=question, transcription=transcription_words
-        )
-
-        model_truth = TranscriptionTruth(
-            _t="TranscriptionTruth",
-            correctWords=true_words,
-            strictGrading=strict_grading,
-        )
-
-        self._rapid_parts.append(
-            ValidatioRapidParts(
-                question=question,
-                asset=asset,
-                payload=payload,
-                truths=model_truth,
-                metadata=metadata,
-                randomCorrectProbability = 1 / len(transcription_words),
-            )
-        )

rapidata/rapidata_client/order/order_builder.py

@@ -1,25 +0,0 @@
-from rapidata.service.openapi_service import OpenAPIService
-from rapidata.rapidata_client.simple_builders.simple_classification_builders import ClassificationQuestionBuilder
-from rapidata.rapidata_client.simple_builders.simple_compare_builders import CompareCriteriaBuilder
-from rapidata.rapidata_client.simple_builders.simple_free_text_builders import FreeTextQuestionBuilder
-from rapidata.rapidata_client.simple_builders.simple_select_words_builders import SelectWordsInstructionBuilder
-from rapidata.rapidata_client.order.rapidata_order_builder import RapidataOrderBuilder
-
-class BaseOrderBuilder():
-    def __init__(self, openapi_service: OpenAPIService):
-        self.openapi_service = openapi_service
-    
-    def classify_order(self, name: str):
-        return ClassificationQuestionBuilder(name, self.openapi_service)
-    
-    def compare_order(self, name: str):
-        return CompareCriteriaBuilder(name, self.openapi_service)
-
-    def free_text_order(self, name: str):
-        return FreeTextQuestionBuilder(name, self.openapi_service)
-    
-    def select_words_order(self, name: str):
-        return SelectWordsInstructionBuilder(name, self.openapi_service)
-
-    def advanced_order(self, name: str):
-        return RapidataOrderBuilder(name, self.openapi_service)