[refactor] finally (i hope) fixed mr issues

Author: domosedy

src/pysatl_core/distributions/support.py

@@ -14,13 +14,20 @@
 __copyright__ = "Copyright (c) 2025 PySATL project"
 __license__ = "SPDX-License-Identifier: MIT"
 
+from collections.abc import Callable
 from dataclasses import dataclass
 from math import floor
-from typing import TYPE_CHECKING, Protocol, cast, overload, runtime_checkable
+from typing import (
+    TYPE_CHECKING,
+    Protocol,
+    cast,
+    overload,
+    runtime_checkable,
+)
 
 import numpy as np
 
-from pysatl_core.types import BoolArray, Interval1D, Number, NumericArray
+from pysatl_core.types import BoolArray, Interval1D, IntervalND, Number, NumericArray
 
 if TYPE_CHECKING:
     from collections.abc import Iterable, Iterator
@@ -49,6 +56,15 @@ class ContinuousSupport(Interval1D, Support):
     """
 
 
+class ContinuousNDSupport(IntervalND, Support):  # type: ignore[misc]
+    """
+    Support for continuous distributions represented as an array of intervals.
+
+    This class inherits from IntervalND and implements the Support protocol
+    for continuous distributions defined on a list of intervals [left, right].
+    """
+
+
 @runtime_checkable
 class DiscreteSupport(Support, Protocol):
     """
@@ -430,10 +446,26 @@ def is_right_bounded(self) -> bool:
     __iter__ = iter_points
 
 
+class SupportByPredicate:
+    def __init__(self, predicate: Callable[[NumericArray | Number], bool]):
+        self._predicate = predicate
+
+    def __contains__(self, item: NumericArray | Number) -> bool:
+        return self._predicate(item)
+
+
+class SupportByIntervals(SupportByPredicate):
+    def __init__(self, support: ContinuousNDSupport):
+        SupportByPredicate.__init__(self, lambda x: x in support)
+
+
 __all__ = [
     # Base support protocol
     "Support",
     "ContinuousSupport",
+    "ContinuousNDSupport",
+    "SupportByPredicate",
+    "SupportByIntervals",
     # Discrete support protocol and implementations
     "DiscreteSupport",
     "ExplicitTableDiscreteSupport",

src/pysatl_core/families/__init__.py

@@ -15,12 +15,10 @@
 from .configuration import configure_families_register
 from .distribution import ParametricFamilyDistribution
 from .exponential_family import (
+    # CanonicalContinuousExponentialClassFamily,
+    ContinuousExponentialClassFamily,
     ExponentialConjugateHyperparameters,
-    ExponentialFamily,
     ExponentialFamilyParametrization,
-    NaturalExponentialFamily,
-    SpacePredicate,
-    SpacePredicateArray,
 )
 from .parametric_family import ParametricFamily
 from .parametrizations import (
@@ -42,12 +40,10 @@
     "configure_families_register",
     # builtins
     *_builtins_all,
-    "ExponentialFamily",
+    "ContinuousExponentialClassFamily",
     "ExponentialFamilyParametrization",
     "ExponentialConjugateHyperparameters",
-    "SpacePredicate",
-    "SpacePredicateArray",
-    "NaturalExponentialFamily",
+    # "CanonicalContinuousExponentialClassFamily",
 ]
 
 del _builtins_all