test: improve coverage

aphp · Feb 17, 2024 · 85b02eb · 85b02eb
1 parent 3175775
commit 85b02eb
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 12 deletions.
diff --git a/edspdf/pipeline.py b/edspdf/pipeline.py
@@ -360,7 +360,7 @@ def to_doc(doc):
                 backend = accelerator
             elif isinstance(accelerator, dict):
                 kwargs = dict(accelerator)
-                backend = accelerator.pop("@accelerator", "simple")
+                backend = kwargs.pop("@accelerator", "simple")
             elif "Accelerator" in type(accelerator).__name__:
                 backend = (
                     "multiprocessing"

diff --git a/edspdf/trainable_pipe.py b/edspdf/trainable_pipe.py
@@ -203,11 +203,6 @@ def named_component_children(self):
             if isinstance(module, TrainablePipe):
                 yield name, module
 
-    def named_component_modules(self):
-        for name, module in self.named_modules():
-            if isinstance(module, TrainablePipe):
-                yield name, module
-
     def post_init(self, gold_data: Iterable[PDFDoc], exclude: Set[str]):
         """
         This method completes the attributes of the component, by looking at some

diff --git a/tests/core/test_pipeline.py b/tests/core/test_pipeline.py
@@ -346,12 +346,13 @@ def error_pipe(doc: PDFDoc):
 
 def test_deprecated_multiprocessing_gpu_stub(frozen_pipeline, pdf, letter_pdf):
     edspdf.accelerators.multiprocessing.MAX_NUM_PROCESSES = 2
-    accelerator = edspdf.accelerators.multiprocessing.MultiprocessingAccelerator(
-        batch_size=2,
-        num_gpu_workers=1,
-        num_cpu_workers=1,
-        gpu_worker_devices=["cpu"],
-    )
+    accelerator = {
+        "@accelerator": "multiprocessing",
+        "batch_size": 2,
+        "num_gpu_workers": 1,
+        "num_cpu_workers": 1,
+        "gpu_worker_devices": ["cpu"],
+    }
     list(
         frozen_pipeline.pipe(
             chain.from_iterable(