arthur-common 2.1.52__py3-none-any.whl → 2.1.54__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of arthur-common might be problematic. Click here for more details.

Files changed (23) hide show
  1. arthur_common/aggregations/aggregator.py +6 -0
  2. arthur_common/aggregations/functions/agentic_aggregations.py +875 -0
  3. arthur_common/aggregations/functions/categorical_count.py +14 -1
  4. arthur_common/aggregations/functions/confusion_matrix.py +35 -5
  5. arthur_common/aggregations/functions/inference_count.py +14 -1
  6. arthur_common/aggregations/functions/inference_count_by_class.py +23 -1
  7. arthur_common/aggregations/functions/inference_null_count.py +15 -1
  8. arthur_common/aggregations/functions/mean_absolute_error.py +26 -3
  9. arthur_common/aggregations/functions/mean_squared_error.py +26 -3
  10. arthur_common/aggregations/functions/multiclass_confusion_matrix.py +43 -5
  11. arthur_common/aggregations/functions/multiclass_inference_count_by_class.py +14 -1
  12. arthur_common/aggregations/functions/numeric_stats.py +14 -1
  13. arthur_common/aggregations/functions/numeric_sum.py +15 -1
  14. arthur_common/aggregations/functions/shield_aggregations.py +127 -16
  15. arthur_common/models/datasets.py +1 -0
  16. arthur_common/models/metrics.py +35 -18
  17. arthur_common/models/schema_definitions.py +58 -0
  18. arthur_common/models/shield.py +158 -0
  19. arthur_common/models/task_job_specs.py +26 -2
  20. arthur_common/tools/aggregation_analyzer.py +2 -1
  21. {arthur_common-2.1.52.dist-info → arthur_common-2.1.54.dist-info}/METADATA +1 -1
  22. {arthur_common-2.1.52.dist-info → arthur_common-2.1.54.dist-info}/RECORD +23 -22
  23. {arthur_common-2.1.52.dist-info → arthur_common-2.1.54.dist-info}/WHEEL +0 -0
arthur_common/aggregations/aggregator.py CHANGED
@@ -29,6 +29,12 @@ class AggregationFunction(ABC):
29
29
  def aggregation_type(self) -> Type[SketchMetric] | Type[NumericMetric]:
30
30
  raise NotImplementedError
31
31
 
32
+ @staticmethod
33
+ @abstractmethod
34
+ def reported_aggregations() -> list[BaseReportedAggregation]:
35
+ """Returns the list of aggregations reported by the aggregate function."""
36
+ raise NotImplementedError
37
+
32
38
  @abstractmethod
33
39
  def aggregate(
34
40
  self,