[skip ci] Add missing exports

mdbenito · mdbenito · commit 84c827bed885 · 2025-09-07T15:22:08.000+02:00
diff --git a/src/pydvl/influence/influence_calculator.py b/src/pydvl/influence/influence_calculator.py
@@ -23,6 +23,9 @@
     "DaskInfluenceCalculator",
     "SequentialInfluenceCalculator",
     "DisableClientSingleThreadCheck",
+    "InvalidDimensionChunksError",
+    "UnalignedChunksError",
+    "ThreadSafetyViolationError",
 ]
 
 logger = logging.getLogger(__name__)
diff --git a/src/pydvl/influence/torch/util.py b/src/pydvl/influence/torch/util.py
@@ -59,6 +59,8 @@
     "safe_torch_linalg_eigh",
     "SecondOrderMode",
     "get_model_parameters",
+    "TorchCatAggregator",
+    "NestedTorchCatAggregator",
 ]
 
 

Original file line number	Diff line number	Diff line change
`@@ -23,6 +23,9 @@`
`23`	`23`	`"DaskInfluenceCalculator",`
`24`	`24`	`"SequentialInfluenceCalculator",`
`25`	`25`	`"DisableClientSingleThreadCheck",`
	`26`	`+ "InvalidDimensionChunksError",`
	`27`	`+ "UnalignedChunksError",`
	`28`	`+ "ThreadSafetyViolationError",`
`26`	`29`	`]`
`27`	`30`
`28`	`31`	`logger = logging.getLogger(__name__)`