3906232596

Build Type

push

github

Committed by Artur Barseghyan

Commit Message

Mention ODT file support in documentation. Add tests. Add ODT file to AugmentFileFromDir provider.

Run Details

1022 of 1024 relevant lines covered (99.8%)

6.46 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

100.0

/src/faker_file/providers/augment_file_from_dir/augmenters/nlpaug_augmenter.py

import nlpaug.augmenter.word as naw

from .base import BaseTextAugmenter

__author__ = "Artur Barseghyan <artur.barseghyan@gmail.com>"
__copyright__ = "2022-2023 Artur Barseghyan"
__license__ = "MIT"
__all__ = ("ContextualWordEmbeddingsAugmenter",)


class ContextualWordEmbeddingsAugmenter(BaseTextAugmenter):
    """Text extractor based on `ContextualWordEmbsAug` of `nlpaug`.

    Usage example:

        from faker import Faker
        from faker_file.providers.augment_file_from_dir import (
            AugmentFileFromDirProvider,
        )
        from faker_file.providers.augment_file_from_dir.augmenters import (
            nlpaug_augmenter,
        )

        FAKER = Faker()

        file = AugmentFileFromDirProvider(FAKER).augment_file_from_dir(
            text_augmenter_cls=nlpaug_augmenter.NlpAugContextualWordEmbsAug
        )
    """

    def handle_kwargs(
        self: "ContextualWordEmbeddingsAugmenter", **kwargs
    ) -> None:
        """Handle kwargs."""

    def augment(
        self: "ContextualWordEmbeddingsAugmenter",
        text: str,
    ) -> str:
        """Augment text."""
        aug = naw.ContextualWordEmbsAug(
            model_path="bert-base-cased",
            action="substitute",
        )
        return aug.augment(text)[0]

1	import nlpaug.augmenter.word as naw	4✔
2
3	from .base import BaseTextAugmenter	4✔
4
5	__author__ = "Artur Barseghyan <artur.barseghyan@gmail.com>"	4✔
6	__copyright__ = "2022-2023 Artur Barseghyan"	4✔
7	__license__ = "MIT"	4✔
8	__all__ = ("ContextualWordEmbeddingsAugmenter",)	4✔
9
10
11	class ContextualWordEmbeddingsAugmenter(BaseTextAugmenter):	4✔
12	"""Text extractor based on `ContextualWordEmbsAug` of `nlpaug`.
13
14	Usage example:
15
16	from faker import Faker
17	from faker_file.providers.augment_file_from_dir import (
18	AugmentFileFromDirProvider,
19	)
20	from faker_file.providers.augment_file_from_dir.augmenters import (
21	nlpaug_augmenter,
22	)
23
24	FAKER = Faker()
25
26	file = AugmentFileFromDirProvider(FAKER).augment_file_from_dir(
27	text_augmenter_cls=nlpaug_augmenter.NlpAugContextualWordEmbsAug
28	)
29	"""
30
31	def handle_kwargs(	4✔
32	self: "ContextualWordEmbeddingsAugmenter", **kwargs
33	) -> None:
34	"""Handle kwargs."""
35
36	def augment(	4✔
37	self: "ContextualWordEmbeddingsAugmenter",
38	text: str,
39	) -> str:
40	"""Augment text."""
41	aug = naw.ContextualWordEmbsAug(	4✔
42	model_path="bert-base-cased",
43	action="substitute",
44	)
45	return aug.augment(text)[0]	4✔

barseghyanartur / faker-file / 3906232596

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous