arthur-client 0.4.1__tar.gz → 0.6.0__tar.gz

{arthur-client-0.4.1 → arthur-client-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arthur-client
-Version: 0.4.1
+Version: 0.6.0
 Summary: Arthur API Python Client
 Author-email: Arthur <info@arthur.ai>
 License: MIT

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur/client/rest/admin/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  admin.yaml
-#   timestamp: 2023-09-26T01:39:46+00:00
+#   timestamp: 2023-10-04T15:04:36+00:00
 
 from __future__ import annotations
 

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur/client/rest/alerts/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  alerts.yaml
-#   timestamp: 2023-09-26T01:39:45+00:00
+#   timestamp: 2023-10-04T15:04:36+00:00
 
 from __future__ import annotations
 

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur/client/rest/bench/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  bench.yaml
-#   timestamp: 2023-09-26T01:39:45+00:00
+#   timestamp: 2023-10-04T15:04:36+00:00
 
 from __future__ import annotations
 

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur/client/rest/common/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  common.yaml
-#   timestamp: 2023-09-26T01:39:45+00:00
+#   timestamp: 2023-10-04T15:04:36+00:00
 
 from __future__ import annotations
 

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur/client/rest/enrichments/client.py

@@ -8,18 +8,20 @@ from arthur.client.http.requests import HTTPClient
 
 from arthur.client.rest.enrichments.models import (
     AnomalyDetectionEnrichmentConfiguration,
+    AnomalyDetectionEnrichmentResponse,
     BiasConstraintEnum,
     BiasMitigationEnrichmentConfiguration,
-    EnrichmentName,
-    EnrichmentStatusUpdate,
-    EnrichmentsConfiguration,
-    ExplainabilityEnrichmentConfiguration,
+    BiasMitigationEnrichmentResponse,
+    EnrichmentsRequest,
+    EnrichmentsResponse,
     ExplainabilityEnrichmentMultipartRequestBody,
     ExplainabilityEnrichmentRequest,
+    ExplainabilityEnrichmentResponse,
     ExplainabilityResultOnDemand,
     ExplainabilityResultWhatIf,
     FindHotspotsResponse,
     HotspotsEnrichmentConfiguration,
+    HotspotsEnrichmentResponse,
     PaginatedBiasMitigationCurves,
     WhatIfRequest,
 )
@@ -85,7 +87,7 @@ class ArthurEnrichmentsClient:
         )
         return PaginatedBiasMitigationCurves(**parsed_resp)
 
-    def get_enrichment_configuration(self, model_id: str) -> EnrichmentsConfiguration:
+    def get_enrichment_configuration(self, model_id: str) -> EnrichmentsResponse:
         """
         Gets the enrichment configurations for a model
 
@@ -95,19 +97,19 @@ class ArthurEnrichmentsClient:
         parsed_resp: Dict[str, Any] = self.http_client.get(  # type: ignore
             f"/v3/models/{model_id}/enrichments", validation_response_code=200
         )
-        return EnrichmentsConfiguration(**parsed_resp)
+        return EnrichmentsResponse(**parsed_resp)
 
     def update_enrichment_config(
         self,
         model_id: str,
-        json_body: EnrichmentsConfiguration,
+        json_body: EnrichmentsRequest,
         multipart_data: ExplainabilityEnrichmentMultipartRequestBody,
     ) -> Response:
         """
         Updates the enrichment configuration for a model
 
         :param model_id:
-        :param json_body: A JSON-formatted enrichments configuration. See the EnrichmentsConfiguration object for schema
+        :param json_body: Configures multiple enrichments. A multipart/form-data body with at least a `configuration` JSON body. If explainability is being enabled for the first time, artifacts must be supplied.
         :param multipart_data: When setting up explainability, a config must always be provided. The explainability enrichment artifact files may be provided all together, but a config must be provided as well, regardless of whether the config has already been set.
         """
 
@@ -122,7 +124,7 @@ class ArthurEnrichmentsClient:
 
     def get_anomaly_detection_config(
         self, model_id: str
-    ) -> AnomalyDetectionEnrichmentConfiguration:
+    ) -> AnomalyDetectionEnrichmentResponse:
         """
 
 
@@ -133,7 +135,7 @@ class ArthurEnrichmentsClient:
             f"/v3/models/{model_id}/enrichments/anomaly_detection",
             validation_response_code=200,
         )
-        return AnomalyDetectionEnrichmentConfiguration(**parsed_resp)
+        return AnomalyDetectionEnrichmentResponse(**parsed_resp)
 
     def update_anomaly_detection_config(
         self, model_id: str, json_body: AnomalyDetectionEnrichmentConfiguration
@@ -155,7 +157,7 @@ class ArthurEnrichmentsClient:
 
     def get_bias_mitigation_config(
         self, model_id: str
-    ) -> BiasMitigationEnrichmentConfiguration:
+    ) -> BiasMitigationEnrichmentResponse:
         """
 
 
@@ -166,7 +168,7 @@ class ArthurEnrichmentsClient:
             f"/v3/models/{model_id}/enrichments/bias_mitigation",
             validation_response_code=200,
         )
-        return BiasMitigationEnrichmentConfiguration(**parsed_resp)
+        return BiasMitigationEnrichmentResponse(**parsed_resp)
 
     def update_bias_mitigation_config(
         self, model_id: str, json_body: BiasMitigationEnrichmentConfiguration
@@ -186,7 +188,7 @@ class ArthurEnrichmentsClient:
         )
         return raw_resp
 
-    def get_hotspots_config(self, model_id: str) -> HotspotsEnrichmentConfiguration:
+    def get_hotspots_config(self, model_id: str) -> HotspotsEnrichmentResponse:
         """
         Get hotspot enrichment config for a model
 
@@ -196,7 +198,7 @@ class ArthurEnrichmentsClient:
         parsed_resp: Dict[str, Any] = self.http_client.get(  # type: ignore
             f"/v3/models/{model_id}/enrichments/hotspots", validation_response_code=200
         )
-        return HotspotsEnrichmentConfiguration(**parsed_resp)
+        return HotspotsEnrichmentResponse(**parsed_resp)
 
     def update_hotspots_config(
         self, model_id: str, json_body: HotspotsEnrichmentConfiguration
@@ -251,7 +253,7 @@ class ArthurEnrichmentsClient:
 
     def get_explainability_config(
         self, model_id: str
-    ) -> ExplainabilityEnrichmentConfiguration:
+    ) -> ExplainabilityEnrichmentResponse:
         """
 
 
@@ -262,7 +264,7 @@ class ArthurEnrichmentsClient:
             f"/v3/models/{model_id}/enrichments/explainability",
             validation_response_code=200,
         )
-        return ExplainabilityEnrichmentConfiguration(**parsed_resp)
+        return ExplainabilityEnrichmentResponse(**parsed_resp)
 
     def update_explainability_config(
         self,
@@ -287,28 +289,6 @@ class ArthurEnrichmentsClient:
         )
         return raw_resp
 
-    def update_enrichment_status(
-        self,
-        model_id: str,
-        enrichment_name: EnrichmentName,
-        json_body: EnrichmentStatusUpdate,
-    ) -> Response:
-        """
-        Update status for an enrichment
-
-        :param model_id:
-        :param enrichment_name: Name of enrichment.
-        :param json_body: Updates status for an enrichment. A body with at least a `status` key.
-        """
-
-        raw_resp: Response = self.http_client.patch(  # type: ignore
-            f"/v3/models/{model_id}/enrichments/{enrichment_name}/status",
-            json=json_body.dict(by_alias=True, exclude_none=True),
-            validation_response_code=202,
-            return_raw_response=True,
-        )
-        return raw_resp
-
     def explain_inference(
         self,
         model_id: str,

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur/client/rest/enrichments/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  enrichments.yaml
-#   timestamp: 2023-09-26T01:39:45+00:00
+#   timestamp: 2023-10-04T15:04:36+00:00
 
 from __future__ import annotations
 
@@ -12,132 +12,164 @@ from arthur.client.types import ByteField
 from pydantic import BaseModel, Extra, Field
 
 
-class BiasMitigationEnrichmentConfiguration(BaseModel):
-    enabled: bool
+class Status(str, Enum):
     """
-    Whether or not the enrichment is enabled
+    The enrichment pipeline status
     """
 
-
-class ExplanationValuesWhatIf(BaseModel):
-    attribute_name: str = Field(..., example='feature_a')
-    explanation_value: float = Field(..., example=0.12)
+    Disabled = 'Disabled'
+    Pending = 'Pending'
+    Training = 'Training'
+    Ready = 'Ready'
+    Failed = 'Failed'
 
 
-class PipelineStatus(str, Enum):
+class EnrichmentStatus(BaseModel):
+    status: Optional[Status] = Field(None, example='Disabled')
     """
-    status of the enrichment pipeline, the status will be 'updating' while enrichments are being reconfigured, and 'ready' or 'disabled' when the updates are complete.
+    The enrichment pipeline status
     """
 
-    Disabled = 'disabled'
-    Updating = 'updating'
-    Ready = 'ready'
 
-
-class EnrichmentStatus(BaseModel):
-    pipeline_status: Optional[PipelineStatus] = None
+class AnomalyDetectionEnrichmentConfiguration(BaseModel):
+    enabled: bool
     """
-    status of the enrichment pipeline, the status will be 'updating' while enrichments are being reconfigured, and 'ready' or 'disabled' when the updates are complete.
+    Whether or not the enrichment is enabled
     """
 
 
-class AnomalyDetectionEnrichmentConfiguration(BiasMitigationEnrichmentConfiguration):
+class BiasMitigationEnrichmentConfiguration(AnomalyDetectionEnrichmentConfiguration):
     pass
 
 
-class FindHotspotsNode(BaseModel):
-    left: Optional[FindHotspotsNode] = None
-    right: Optional[FindHotspotsNode] = None
-    rules: Dict[str, Any]
+class ExplanationAlgo(str, Enum):
     """
-    rules for the split on this node
+    Explainability algorithm to use in the model server. Current options are "lime" or "shap", default is "lime"
     """
-    gt_to_info: Optional[Dict[str, Any]] = None
+
+    LIME = 'lime'
+    SHAP = 'shap'
+
+
+class Config(BaseModel):
+    python_version: Optional[str] = Field(None, example='3.8.17')
     """
-    info around ground truths at this node
+    Python version number user project users
     """
-    precision: Optional[float] = None
+    sdk_version: Optional[str] = Field(None, example='3.26.0')
     """
-    precision for this node
+    SDK version number user project uses
     """
-    recall: Optional[float] = None
+    streaming_explainability_enabled: Optional[bool] = Field(None, example=True)
     """
-    recall for this node
+    Set to true if explainability should be calculated for all inferences in a streaming manner and set to false if explainability should only be calculated when requested.
     """
-    f1: Optional[float] = None
+    user_predict_function_import_path: Optional[str] = Field(
+        None, example='path/to/import/function.py'
+    )
     """
-    f1 for this node
+    Import path of the user model predict function in the project directory
     """
-    accuracy: float
+    shap_expected_values: Optional[List[float]] = Field(None, example=[1, 2, 3])
     """
-    accuracy for this node
+    If using SHAP, these are the expected values generated when the explainer is created
     """
-    impurity: float
+    model_server_cpu: Optional[str] = Field('2', example='2')
     """
-    impurity for this node
+    Number of cpus to assign to explanation server.
     """
-    n_samples: int
+    model_server_memory: Optional[str] = Field('1500Mi', example='1Gi')
     """
-    n_samples used for this node
+    Amount of memory to assign explanation server in the format xMi or xGi.
     """
-    feature: str
+    model_server_max_replicas: Optional[int] = Field(30, example=30)
     """
-    name of feature this node was cut on
+    Max number of model server instances.
     """
-    cutoff: Optional[float] = None
+    explanation_nsamples: Optional[int] = Field(2000, example=2000)
     """
-    the cutoff for the node
+    Explainability algorithms create sample data points when explaining inferences. The number of samples created per explanation can be configured here. There is a trade off between accuracy and computing power and time for this configuration.
     """
-
-
-class AnomalyDetectionEnrichmentResponse(AnomalyDetectionEnrichmentConfiguration):
+    explanation_algo: Optional[ExplanationAlgo] = Field('lime', example='lime')
     """
-    The response object containing configuration and status of an on-by-default enrichment.
+    Explainability algorithm to use in the model server. Current options are "lime" or "shap", default is "lime"
+    """
+    inference_consumer_cpu: Optional[str] = Field('500m', example='500m')
+    """
+    Number of cpus to assign to the inference reader
+    """
+    inference_consumer_memory: Optional[str] = Field('512Mi', example='512Mi')
+    """
+    Amount of memory to assign to the inference reader in the format xMi or xGi.
+    """
+    inference_consumer_score_percent: Optional[float] = Field(1.0, example=0.1)
+    """
+    Sampling rate for inferences to explain
+    """
+    inference_consumer_thread_pool_size: Optional[int] = Field(5, example=5)
+    """
+    Number of threads in the inference consumer pool
     """
-
-    pass
 
 
-class BiasMitigationEnrichmentResponse(BiasMitigationEnrichmentConfiguration):
+class ExplainabilityEnrichmentConfiguration(BaseModel):
     """
-    The response object containing configuration and status of an on-by-default enrichment.
+    JSON-formatted configuration options for the Explainability Enrichment. See the ExplainabilityEnrichmentConfiguration for schema.
     """
 
-    pass
+    enabled: Optional[bool] = Field(None, example=True)
+    """
+    Whether or not the enrichment is enabled
+    """
+    config: Optional[Config] = None
 
 
-class HotspotsEnrichmentConfiguration(BiasMitigationEnrichmentConfiguration):
+class HotspotsEnrichmentConfiguration(AnomalyDetectionEnrichmentConfiguration):
     pass
 
 
-class ExpectedValues(BaseModel):
-    predicted_attribute_name: str = Field(..., example='feature_a')
-    expected_value: float = Field(..., example=0.12)
-
-
-class TokenObject(BaseModel):
-    token: str = Field(..., example='dog')
+class EnrichmentsConfiguration(BaseModel):
     """
-    Token string which is generated from separating the input text by the model's given delimiter.
+    A JSON-formatted enrichments configuration. See the EnrichmentsConfiguration object for schema
     """
-    position: float = Field(..., example=0)
+
+    anomaly_detection: Optional[AnomalyDetectionEnrichmentConfiguration] = None
+    bias_mitigation: Optional[BiasMitigationEnrichmentConfiguration] = None
+    hotspots: Optional[HotspotsEnrichmentConfiguration] = None
+    explainability: Optional[ExplainabilityEnrichmentConfiguration] = None
+
+
+class AnomalyDetectionEnrichmentResponse(
+    AnomalyDetectionEnrichmentConfiguration, EnrichmentStatus
+):
     """
-    Integer representing the location of the token in the input text. 0 refers to the the first token in the input text.
+    The response object containing configuration and status of an on-by-default enrichment.
     """
-    explanation_value: float = Field(..., example=0.48)
+
+    pass
+
+
+class BiasMitigationEnrichmentResponse(
+    BiasMitigationEnrichmentConfiguration, EnrichmentStatus
+):
     """
-    Float explanation value for the specific token.
+    The response object containing configuration and status of an on-by-default enrichment.
     """
 
+    pass
 
-class FindHotspotsResponse(BaseModel):
-    data: List[FindHotspotsNode]
+
+class ExplainabilityEnrichmentResponse(
+    ExplainabilityEnrichmentConfiguration, EnrichmentStatus
+):
     """
-    Contains all hotspots based on input
+    The response object containing configuration and status of the explainability enrichment.
     """
 
+    pass
+
 
-class HotspotsEnrichmentResponse(HotspotsEnrichmentConfiguration):
+class HotspotsEnrichmentResponse(HotspotsEnrichmentConfiguration, EnrichmentStatus):
     """
     The response object containing configuration and status of an on-by-default enrichment.
     """
@@ -145,93 +177,121 @@ class HotspotsEnrichmentResponse(HotspotsEnrichmentConfiguration):
     pass
 
 
-class ExplanationInput(BaseModel):
-    class Config:
-        extra = Extra.allow
+class EnrichmentsResponse(BaseModel):
+    """
+    The response object containing configuration and status of all enrichments.
+    """
 
-    __root__: Optional[Dict[str, Dict[str, Any]]] = None
+    anomaly_detection: Optional[AnomalyDetectionEnrichmentResponse] = None
+    bias_mitigation: Optional[BiasMitigationEnrichmentResponse] = None
+    hotspots: Optional[HotspotsEnrichmentResponse] = None
+    explainability: Optional[ExplainabilityEnrichmentResponse] = None
 
 
-class ExplanationAlgo(str, Enum):
+class ExplainabilityEnrichmentRequest(BaseModel):
     """
-    Explainability algorithm to use in the model server. Current options are "lime" or "shap", default is "lime"
+    Configures explainability. A multipart/form-data body with at least a `configuration` JSON body. If explainability is being enabled for the first time, artifacts must be supplied.
     """
 
-    LIME = 'lime'
-    SHAP = 'shap'
+    config: Optional[ExplainabilityEnrichmentConfiguration] = None
+    """
+    Explainability enrichment configuration
+    """
 
 
-class Config(BaseModel):
-    python_version: Optional[str] = Field(None, example='3.8.17')
+class EnrichmentsRequest(BaseModel):
     """
-    Python version number user project users
+    Configures multiple enrichments. A multipart/form-data body with at least a `configuration` JSON body. If explainability is being enabled for the first time, artifacts must be supplied.
     """
-    sdk_version: Optional[str] = Field(None, example='3.26.0')
+
+    config: Optional[EnrichmentsConfiguration] = None
     """
-    SDK version number user project uses
+    Enrichments configuration
     """
-    streaming_explainability_enabled: Optional[bool] = Field(None, example=True)
+
+
+class ExplanationValuesWhatIf(BaseModel):
+    attribute_name: str = Field(..., example='feature_a')
+    explanation_value: float = Field(..., example=0.12)
+
+
+class FindHotspotsNode(BaseModel):
+    left: Optional[FindHotspotsNode] = None
+    right: Optional[FindHotspotsNode] = None
+    rules: Dict[str, Any]
     """
-    Set to true if explainability should be calculated for all inferences in a streaming manner and set to false if explainability should only be calculated when requested.
+    rules for the split on this node
     """
-    user_predict_function_import_path: Optional[str] = Field(
-        None, example='path/to/import/function.py'
-    )
+    gt_to_info: Optional[Dict[str, Any]] = None
     """
-    Import path of the user model predict function in the project directory
+    info around ground truths at this node
     """
-    shap_expected_values: Optional[List[float]] = Field(None, example=[1, 2, 3])
+    precision: Optional[float] = None
     """
-    If using SHAP, these are the expected values generated when the explainer is created
+    precision for this node
     """
-    model_server_cpu: Optional[str] = Field('2', example='2')
+    recall: Optional[float] = None
     """
-    Number of cpus to assign to explanation server.
+    recall for this node
     """
-    model_server_memory: Optional[str] = Field('1500Mi', example='1Gi')
+    f1: Optional[float] = None
     """
-    Amount of memory to assign explanation server in the format xMi or xGi.
+    f1 for this node
     """
-    model_server_max_replicas: Optional[int] = Field(30, example=30)
+    accuracy: float
     """
-    Max number of model server instances.
+    accuracy for this node
     """
-    explanation_nsamples: Optional[int] = Field(2000, example=2000)
+    impurity: float
     """
-    Explainability algorithms create sample data points when explaining inferences. The number of samples created per explanation can be configured here. There is a trade off between accuracy and computing power and time for this configuration.
+    impurity for this node
     """
-    explanation_algo: Optional[ExplanationAlgo] = Field('lime', example='lime')
+    n_samples: int
     """
-    Explainability algorithm to use in the model server. Current options are "lime" or "shap", default is "lime"
+    n_samples used for this node
     """
-    inference_consumer_cpu: Optional[str] = Field('500m', example='500m')
+    feature: str
     """
-    Number of cpus to assign to the inference reader
+    name of feature this node was cut on
     """
-    inference_consumer_memory: Optional[str] = Field('512Mi', example='512Mi')
+    cutoff: Optional[float] = None
     """
-    Amount of memory to assign to the inference reader in the format xMi or xGi.
+    the cutoff for the node
     """
-    inference_consumer_score_percent: Optional[float] = Field(1.0, example=0.1)
+
+
+class ExpectedValues(BaseModel):
+    predicted_attribute_name: str = Field(..., example='feature_a')
+    expected_value: float = Field(..., example=0.12)
+
+
+class TokenObject(BaseModel):
+    token: str = Field(..., example='dog')
     """
-    Sampling rate for inferences to explain
+    Token string which is generated from separating the input text by the model's given delimiter.
     """
-    inference_consumer_thread_pool_size: Optional[int] = Field(5, example=5)
+    position: float = Field(..., example=0)
     """
-    Number of threads in the inference consumer pool
+    Integer representing the location of the token in the input text. 0 refers to the the first token in the input text.
     """
-
-
-class ExplainabilityEnrichmentConfiguration(BaseModel):
+    explanation_value: float = Field(..., example=0.48)
     """
-    JSON-formatted configuration options for the Explainability Enrichment. See the ExplainabilityEnrichmentConfiguration for schema.
+    Float explanation value for the specific token.
     """
 
-    enabled: Optional[bool] = Field(None, example=True)
+
+class FindHotspotsResponse(BaseModel):
+    data: List[FindHotspotsNode]
     """
-    Whether or not the enrichment is enabled
+    Contains all hotspots based on input
     """
-    config: Optional[Config] = None
+
+
+class ExplanationInput(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    __root__: Optional[Dict[str, Dict[str, Any]]] = None
 
 
 class ExplanationValuesOnDemand(BaseModel):
@@ -284,20 +344,15 @@ class WhatIfRequest(BaseModel):
     model_pipeline_input: List[WhatIfAttributeRequest]
 
 
-class Status(str, Enum):
-    Disabled = 'Disabled'
-    Pending = 'Pending'
-    Training = 'Training'
-    Ready = 'Ready'
-    Failed = 'Failed'
-
-
-class EnrichmentStatusUpdate(BaseModel):
+class EnrichmentStatusUpdate(EnrichmentStatus):
     """
     Updates status for an enrichment. A body with at least a `status` key.
     """
 
-    status: Status
+    status: Status = Field(..., example='Disabled')
+    """
+    The enrichment pipeline status
+    """
 
 
 class EnrichmentName(str, Enum):
@@ -332,17 +387,6 @@ class BiasMitigationDataPoints(BaseModel):
     """
 
 
-class EnrichmentsConfiguration(BaseModel):
-    """
-    A JSON-formatted enrichments configuration. See the EnrichmentsConfiguration object for schema
-    """
-
-    anomaly_detection: Optional[AnomalyDetectionEnrichmentConfiguration] = None
-    bias_mitigation: Optional[BiasMitigationEnrichmentConfiguration] = None
-    hotspots: Optional[HotspotsEnrichmentConfiguration] = None
-    explainability: Optional[ExplainabilityEnrichmentConfiguration] = None
-
-
 class ExplanationsOnDemand(BaseModel):
     algorithm: str = Field(..., example='shap')
     predicted_attribute_name: str = Field(..., example='class_a')
@@ -355,25 +399,6 @@ class BiasConstraintEnum(str, Enum):
     EqualizedOdds = 'equalized_odds'
 
 
-class ExplainabilityEnrichmentResponse(ExplainabilityEnrichmentConfiguration):
-    """
-    The response object containing configuration and status of the explainability enrichment.
-    """
-
-    pass
-
-
-class ExplainabilityEnrichmentRequest(BaseModel):
-    """
-    Configures explainability. A multipart/form-data body with at least a `configuration` JSON body. If explainability is being enabled for the first time, artifacts must be supplied.
-    """
-
-    config: Optional[ExplainabilityEnrichmentConfiguration] = None
-    """
-    Explainability enrichment configuration
-    """
-
-
 class BiasMitigationCurveResponse(BaseModel):
     id: str = Field(..., example='418c6939-8765-40fa-b04e-11ba57b7f21c')
     """
@@ -425,32 +450,11 @@ class BiasMitigationCurveResponse(BaseModel):
     """
 
 
-class EnrichmentsStatus(BaseModel):
-    """
-    The response object containing configuration and status of all enrichments.
-    """
-
-    anomaly_detection: Optional[EnrichmentStatus] = None
-    bias_mitigation: Optional[EnrichmentStatus] = None
-    explainability: Optional[EnrichmentStatus] = None
-
-
 class ExplainabilityResultOnDemand(BaseModel):
     explanation: List[ExplanationsOnDemand]
     expected_value: Optional[List[ExpectedValues]] = None
 
 
-class EnrichmentsResponse(BaseModel):
-    """
-    The response object containing configuration and status of all enrichments.
-    """
-
-    anomaly_detection: Optional[AnomalyDetectionEnrichmentResponse] = None
-    bias_mitigation: Optional[BiasMitigationEnrichmentResponse] = None
-    hotspots: Optional[HotspotsEnrichmentResponse] = None
-    explainability: Optional[ExplainabilityEnrichmentResponse] = None
-
-
 class ExplainabilityResultWhatIf(BaseModel):
     predicted_values: List[WhatIfAttributeRequest]
     explanation: List[ExplanationsWhatIf]
@@ -465,14 +469,6 @@ class ExplainabilityEnrichmentMultipartRequestBody(ExplainabilityEnrichmentArtif
     config: Optional[ExplainabilityEnrichmentConfiguration] = None
 
 
-class EnrichmentsRequest(EnrichmentsConfiguration):
-    """
-    Configures multiple enrichments. A multipart/form-data body with at least a `configuration` JSON body. If explainability is being enabled for the first time, artifacts must be supplied.
-    """
-
-    pass
-
-
 class PaginatedBiasMitigationCurves(BaseModel):
     data: List[BiasMitigationCurveResponse]
     """

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur/client/rest/inferences/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  inferences.yaml
-#   timestamp: 2023-09-26T01:39:45+00:00
+#   timestamp: 2023-10-04T15:04:36+00:00
 
 from __future__ import annotations
 

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur/client/rest/insights/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  insights.yaml
-#   timestamp: 2023-09-26T01:39:45+00:00
+#   timestamp: 2023-10-04T15:04:35+00:00
 
 from __future__ import annotations
 

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur/client/rest/metrics/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  metrics.yaml
-#   timestamp: 2023-09-26T01:39:45+00:00
+#   timestamp: 2023-10-04T15:04:35+00:00
 
 from __future__ import annotations
 
@@ -32,11 +32,7 @@ class CategoricalValueLimits(BaseModel):
 
 
 class MetricGroupBy(BaseModel):
-    """
-    Model attribute to group by. If provided, the results will be broken down by each value or bin in the attribute
-    """
-
-    __root__: List[str] = Field(..., example=['Education'])
+    __root__: str = Field(..., example='Education')
     """
     Model attribute to group by. If provided, the results will be broken down by each value or bin in the attribute
     """

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur/client/rest/model_groups/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  model_groups.yaml
-#   timestamp: 2023-09-26T01:39:45+00:00
+#   timestamp: 2023-10-04T15:04:35+00:00
 
 from __future__ import annotations
 

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur/client/rest/models/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  models.yaml
-#   timestamp: 2023-09-26T01:39:45+00:00
+#   timestamp: 2023-10-04T15:04:35+00:00
 
 from __future__ import annotations
 

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur/client/rest/query/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  query.yaml
-#   timestamp: 2023-09-26T01:39:44+00:00
+#   timestamp: 2023-10-04T15:04:35+00:00
 
 from __future__ import annotations
 

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur/client/version.py

@@ -1,2 +1,2 @@
-__version__ = "0.4.1"
+__version__ = "0.6.0"
 api_spec_version = "3.0.0"

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arthur-client
-Version: 0.4.1
+Version: 0.6.0
 Summary: Arthur API Python Client
 Author-email: Arthur <info@arthur.ai>
 License: MIT

{arthur-client-0.4.1 → arthur-client-0.6.0}/arthur/client/rest/model_groups/client.py

@@ -6,13 +6,13 @@ from requests import Response
 # import http client
 from arthur.client.http.requests import HTTPClient
 
-from arthur.client.rest.models.models import ModelExpand, ModelObject
 from arthur.client.rest.model_groups.models import (
     ModelGroupResponse,
     ModelGroupUpdateRequest,
     PaginatedModelGroupResponse,
     PaginatedModelGroupVersionsResponse,
 )
+from arthur.client.rest.models.models import ModelExpand, ModelObject
 
 
 PATH_PREFIX = "/api"