Module `lore.types.enums`

Expand source code

from enum import Enum


class DatasetSource(str, Enum):
    HUGGING_FACE = "HUGGING_FACE"
    LOCAL = "LOCAL"


class DatasetDataType(str, Enum):
    TEXT = "TEXT"
    IMAGE = "IMAGE"


class ModelSource(str, Enum):
    HUGGING_FACE = "HUGGING_FACE"
    GENAI = "GENAI"


class ExperimentType(str, Enum):
    FINE_TUNING = "FINE_TUNING"
    EVALUATION = "EVALUATION"


class ControllerType(str, Enum):
    CHAT = "chat"
    DATA = "data"
    EMBEDDING = "embedding"


class ControllerSyncType(str, Enum):
    RESTART = "restart"
    GET = "get"


class PEFTType(str, Enum):
    LORA = "lora"


class DeletionStatus(str, Enum):
    DELETING = "deleting"
    DELETED = "deleted"
    DELETION_FAILED = "deletion_failed"


class GPUType(str, Enum):
    A100 = "A100"
    V100 = "V100"
    T4 = "T4"

    @classmethod
    def from_string(cls, string):
        for gpu in cls:
            if gpu.value in string.upper():
                return gpu
        return None

Classes

class ControllerSyncType (*args, **kwds)

str(object='') -> str str(bytes_or_buffer[, encoding[, errors]]) -> str

Create a new string object from the given object. If encoding or errors is specified, then the object must expose a data buffer that will be decoded using the given encoding and error handler. Otherwise, returns the result of object.str() (if defined) or repr(object). encoding defaults to sys.getdefaultencoding(). errors defaults to 'strict'.

Expand source code

class ControllerSyncType(str, Enum):
    RESTART = "restart"
    GET = "get"

Ancestors

builtins.str
enum.Enum

Class variables

var GET
var RESTART

class ControllerType (*args, **kwds)

str(object='') -> str str(bytes_or_buffer[, encoding[, errors]]) -> str

Expand source code

class ControllerType(str, Enum):
    CHAT = "chat"
    DATA = "data"
    EMBEDDING = "embedding"

Ancestors

builtins.str
enum.Enum

Class variables

var CHAT
var DATA
var EMBEDDING

class DatasetDataType (*args, **kwds)

str(object='') -> str str(bytes_or_buffer[, encoding[, errors]]) -> str

Expand source code

class DatasetDataType(str, Enum):
    TEXT = "TEXT"
    IMAGE = "IMAGE"

Ancestors

builtins.str
enum.Enum

Class variables

var IMAGE
var TEXT

class DatasetSource (*args, **kwds)

str(object='') -> str str(bytes_or_buffer[, encoding[, errors]]) -> str

Expand source code

class DatasetSource(str, Enum):
    HUGGING_FACE = "HUGGING_FACE"
    LOCAL = "LOCAL"

Ancestors

builtins.str
enum.Enum

Class variables

var HUGGING_FACE
var LOCAL

class DeletionStatus (*args, **kwds)

str(object='') -> str str(bytes_or_buffer[, encoding[, errors]]) -> str

Expand source code

class DeletionStatus(str, Enum):
    DELETING = "deleting"
    DELETED = "deleted"
    DELETION_FAILED = "deletion_failed"

Ancestors

builtins.str
enum.Enum

Class variables

var DELETED
var DELETING
var DELETION_FAILED

class ExperimentType (*args, **kwds)

str(object='') -> str str(bytes_or_buffer[, encoding[, errors]]) -> str

Expand source code

class ExperimentType(str, Enum):
    FINE_TUNING = "FINE_TUNING"
    EVALUATION = "EVALUATION"

Ancestors

builtins.str
enum.Enum

Class variables

var EVALUATION
var FINE_TUNING

class GPUType (*args, **kwds)

str(object='') -> str str(bytes_or_buffer[, encoding[, errors]]) -> str

Expand source code

class GPUType(str, Enum):
    A100 = "A100"
    V100 = "V100"
    T4 = "T4"

    @classmethod
    def from_string(cls, string):
        for gpu in cls:
            if gpu.value in string.upper():
                return gpu
        return None

Ancestors

builtins.str
enum.Enum

Class variables

var A100
var T4
var V100

Static methods

def from_string(string)

Expand source code

@classmethod
def from_string(cls, string):
    for gpu in cls:
        if gpu.value in string.upper():
            return gpu
    return None

class ModelSource (*args, **kwds)

str(object='') -> str str(bytes_or_buffer[, encoding[, errors]]) -> str

Expand source code

class ModelSource(str, Enum):
    HUGGING_FACE = "HUGGING_FACE"
    GENAI = "GENAI"

Ancestors

builtins.str
enum.Enum

Class variables

var GENAI
var HUGGING_FACE

class PEFTType (*args, **kwds)

str(object='') -> str str(bytes_or_buffer[, encoding[, errors]]) -> str

Expand source code

class PEFTType(str, Enum):
    LORA = "lora"

Ancestors

builtins.str
enum.Enum

Class variables

var LORA