fix: missing parameter from all pruna enums

.github/workflows/tests.yaml

@@ -24,7 +24,7 @@ jobs:
 
     strategy:
       matrix:
-        python-version: ["3.10"]
+        python-version: ["3.11"]
 
     steps:
       - name: Checkout code
@@ -67,7 +67,7 @@ jobs:
 
     strategy:
       matrix:
-        python-version: ["3.10"]
+        python-version: ["3.11"]
 
     env:
       HF_TOKEN: ${{ secrets.HF_TOKEN }}

pyproject.toml

@@ -103,7 +103,7 @@ dependencies = [
     "transformers",
     "pytorch-lightning",
     "huggingface-hub[hf-xet]>=0.30.0",
-    "datasets>=0.34",
+    "datasets>=3.0",
     "numpy>=1.24.4",
     "numpydoc>=1.6.0",
     "diffusers>=0.21.4",

src/pruna/algorithms/base/tags.py

@@ -37,6 +37,9 @@ class AlgorithmTag(Enum):
         The type of the enum.
     start : int
         The start index for auto-numbering enum values.
+    boundary : enum.FlagBoundary or None
+        Boundary handling mode used by the Enum functional API for Flag and
+        IntFlag enums.
     """
 
     QUANTIZER = (

src/pruna/engine/load.py

@@ -565,6 +565,9 @@ class LOAD_FUNCTIONS(Enum):  # noqa: N801
         The type of the enum.
     start : int
         The start index for auto-numbering enum values.
+    boundary : enum.FlagBoundary or None
+        Boundary handling mode used by the Enum functional API for Flag and
+        IntFlag enums.
 
     Examples
     --------

src/pruna/engine/save.py

@@ -528,6 +528,9 @@ class SAVE_FUNCTIONS(Enum):  # noqa: N801
         The type of the enum.
     start : int
         The start index for auto-numbering enum values.
+    boundary : enum.FlagBoundary or None
+        Boundary handling mode used by the Enum functional API for Flag and
+        IntFlag enums.
 
     Examples
     --------

src/pruna/evaluation/metrics/metric_torch.py

@@ -168,6 +168,9 @@ class TorchMetrics(Enum):
         The type of the enum value.
     start : int
         The starting value for the enum.
+    boundary : enum.FlagBoundary or None
+        Boundary handling mode used by the Enum functional API for Flag and
+        IntFlag enums.
     """
 
     fid = (partial(FrechetInceptionDistance), fid_update, "gt_y")