Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

SPARKNLP-669 Adding missing inputAnnotatorTypes #13144

Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion python/sparknlp/annotator/audio/wav2vec2_for_ctc.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@
"""Contains classes concerning Wav2Vec2ForCTC."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class Wav2Vec2ForCTC(AnnotatorModel,
Expand Down
1 change: 0 additions & 1 deletion python/sparknlp/annotator/chunker.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
# limitations under the License.
"""Contains classes for the Chunker."""
from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class Chunker(AnnotatorModel):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
# limitations under the License.

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class AlbertForQuestionAnswering(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes concerning AlbertForSequenceClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class AlbertForSequenceClassification(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for AlbertForTokenClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class AlbertForTokenClassification(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
# limitations under the License.

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class BertForQuestionAnswering(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for BertForSequenceClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class BertForSequenceClassification(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for BertForTokenClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class BertForTokenClassification(AnnotatorModel,
Expand Down
1 change: 0 additions & 1 deletion python/sparknlp/annotator/classifier_dl/classifier_dl.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,6 @@
from sparknlp.annotator.param import EvaluationDLParams, ClassifierEncoder
from sparknlp.base import DocumentAssembler
from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class ClassifierDLApproach(AnnotatorApproach, EvaluationDLParams, ClassifierEncoder):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
# limitations under the License.

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class DeBertaForQuestionAnswering(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for DeBertaForTokenClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class DeBertaForTokenClassification(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
# limitations under the License.

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class DistilBertForQuestionAnswering(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for DistilBertForSequenceClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class DistilBertForSequenceClassification(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for DistilBertForTokenClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class DistilBertForTokenClassification(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
# limitations under the License.

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class LongformerForQuestionAnswering(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for LongformerForSequenceClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class LongformerForSequenceClassification(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for LongformerForTokenClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class LongformerForTokenClassification(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,6 @@
from sparknlp.annotator.param import EvaluationDLParams, ClassifierEncoder
from sparknlp.annotator.classifier_dl import ClassifierDLModel
from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class MultiClassifierDLApproach(AnnotatorApproach, EvaluationDLParams, ClassifierEncoder):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
# limitations under the License.

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class RoBertaForQuestionAnswering(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for RoBertaForSequenceClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class RoBertaForSequenceClassification(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for RoBertaForTokenClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class RoBertaForTokenClassification(AnnotatorModel,
Expand Down
1 change: 0 additions & 1 deletion python/sparknlp/annotator/classifier_dl/sentiment_dl.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@

from sparknlp.annotator.param import EvaluationDLParams, ClassifierEncoder
from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class SentimentDLApproach(AnnotatorApproach, EvaluationDLParams, ClassifierEncoder):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@

from sparknlp.common import *
from sparknlp.annotator.classifier_dl import BertForQuestionAnswering
from sparknlp.common.annotator_type import AnnotatorType


class TapasForQuestionAnswering(BertForQuestionAnswering):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
# limitations under the License.

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class XlmRoBertaForQuestionAnswering(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for XlmRoBertaForSequenceClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class XlmRoBertaForSequenceClassification(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for XlmRoBertaForTokenClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class XlmRoBertaForTokenClassification(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for XlnetForSequenceClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class XlnetForSequenceClassification(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for XlnetForTokenClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class XlnetForTokenClassification(AnnotatorModel,
Expand Down
1 change: 0 additions & 1 deletion python/sparknlp/annotator/coref/spanbert_coref.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for the SpanBertCorefModel."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class SpanBertCorefModel(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@
"""Contains classes concerning ViTForImageClassification."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class ViTForImageClassification(AnnotatorModel,
Expand Down
1 change: 0 additions & 1 deletion python/sparknlp/annotator/dependency/dependency_parser.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for the DependencyParser."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class DependencyParserApproach(AnnotatorApproach):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@


from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class TypedDependencyParserApproach(AnnotatorApproach):
Expand Down
1 change: 0 additions & 1 deletion python/sparknlp/annotator/document_normalizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
# limitations under the License.
"""Contains classes for the DocumentNormalizer"""
from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class DocumentNormalizer(AnnotatorModel):
Expand Down
1 change: 0 additions & 1 deletion python/sparknlp/annotator/embeddings/albert_embeddings.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes concerning AlbertEmbeddings."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class AlbertEmbeddings(AnnotatorModel,
Expand Down
1 change: 0 additions & 1 deletion python/sparknlp/annotator/embeddings/bert_embeddings.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for BertEmbeddings."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class BertEmbeddings(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for BertSentenceEmbeddings."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class BertSentenceEmbeddings(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for CamemBertEmbeddings."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class CamemBertEmbeddings(AnnotatorModel,
Expand Down
1 change: 0 additions & 1 deletion python/sparknlp/annotator/embeddings/chunk_embeddings.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for ChunkEmbeddings"""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class ChunkEmbeddings(AnnotatorModel):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for DistilBertEmbeddings."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class DistilBertEmbeddings(AnnotatorModel,
Expand Down
1 change: 0 additions & 1 deletion python/sparknlp/annotator/embeddings/doc2vec.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for Doc2Vec."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class Doc2VecApproach(AnnotatorApproach, HasStorageRef, HasEnableCachingProperties):
Expand Down
1 change: 0 additions & 1 deletion python/sparknlp/annotator/embeddings/elmo_embeddings.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for ElmoEmbeddings."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class ElmoEmbeddings(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for LongformerEmbeddings."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class LongformerEmbeddings(AnnotatorModel,
Expand Down
1 change: 0 additions & 1 deletion python/sparknlp/annotator/embeddings/roberta_embeddings.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for RoBertaEmbeddings."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class RoBertaEmbeddings(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for RoBertaSentenceEmbeddings."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class RoBertaSentenceEmbeddings(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for SentenceEmbeddings."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class SentenceEmbeddings(AnnotatorModel, HasEmbeddingsProperties, HasStorageRef):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for the UniversalSentenceEncoder."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class UniversalSentenceEncoder(AnnotatorModel,
Expand Down
1 change: 0 additions & 1 deletion python/sparknlp/annotator/embeddings/word2vec.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@


from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class Word2VecApproach(AnnotatorApproach, HasStorageRef, HasEnableCachingProperties):
Expand Down
1 change: 0 additions & 1 deletion python/sparknlp/annotator/embeddings/word_embeddings.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@


from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class WordEmbeddings(AnnotatorApproach, HasEmbeddingsProperties, HasStorage):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for XlmRoBertaEmbeddings."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class XlmRoBertaEmbeddings(AnnotatorModel,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
"""Contains classes for XlmRoBertaSentenceEmbeddings."""

from sparknlp.common import *
from sparknlp.common.annotator_type import AnnotatorType


class XlmRoBertaSentenceEmbeddings(AnnotatorModel,
Expand Down
Loading