from .database.base import Annotatable

from .transcriptomics import Expression

__all__ = ['Protein']

class Protein(Expression, Annotatable):

    def __init__(self, data, transpose, gene_index=None, usecols=None, gene_level=None, sample_level="sample_index",

                 transform_fn=None, dropna=False, npartitions=None, cohort_name=None):

        """

        Args:

            data:

            transpose:

            gene_index:

            usecols:

            gene_level:

            sample_level:

            transform_fn:

            dropna:

            npartitions:

            cohort_name:

        """

        super().__init__(data=data, transpose=transpose, gene_index=gene_index, usecols=usecols,

                         gene_level=gene_level, sample_level=sample_level, transform_fn=transform_fn,

                         dropna=dropna, npartitions=npartitions, cohort_name=cohort_name)

    @classmethod

    def name(cls):

        return cls.__name__