fix: save qualifiers config during to_disk

aphp · Dec 20, 2023 · ffbe555 · ffbe555
1 parent bd8c582
commit ffbe555
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/edsnlp/pipes/trainable/span_qualifier/span_qualifier.py b/edsnlp/pipes/trainable/span_qualifier/span_qualifier.py
@@ -219,7 +219,7 @@ def to_disk(self, path, *, exclude=set()):
                 },
                 f,
             )
-        super().to_disk(path, exclude=exclude)
+        return super().to_disk(path, exclude=exclude)
 
     def from_disk(self, path, exclude=tuple()):
         repr_id = object.__repr__(self)