Skip to content

Status

ModelStatus

Bases: Status

Status class for model entities.

Source code in digitalhub_ml/entities/models/status.py 
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
class ModelStatus(Status):
    """
    Status class for model entities.
    """

    def __init__(
        self,
        state: str,
        message: str | None = None,
        files: list[dict] | None = None,
        **kwargs,
    ) -> None:
        """
        Constructor.

        Parameters
        ----------
        **kwargs : dict
            Keywords arguments.
        """
        super().__init__(state, message)
        self.files = files

    def add_file(self, file: dict) -> None:
        """
        Add a file to the status.

        Parameters
        ----------
        file : dict
            File to add.

        Returns
        -------
        None
        """

        # Add the file to the list
        if self.files is None:
            self.files = []

        # Remove the file info if it already exists
        self.files = [f for f in self.files if f["path"] != file["path"]]

        # Add the new file
        self.files.append(file)

__init__(state, message=None, files=None, **kwargs)

Constructor.

Parameters:

Name Type Description Default
**kwargs dict

Keywords arguments.

 {}
Source code in digitalhub_ml/entities/models/status.py 
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
def __init__(
    self,
    state: str,
    message: str | None = None,
    files: list[dict] | None = None,
    **kwargs,
) -> None:
    """
    Constructor.

    Parameters
    ----------
    **kwargs : dict
        Keywords arguments.
    """
    super().__init__(state, message)
    self.files = files

add_file(file)

Add a file to the status.

Parameters:

Name Type Description Default
file dict

File to add.

 required

Returns:

Type Description
None
Source code in digitalhub_ml/entities/models/status.py 
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
def add_file(self, file: dict) -> None:
    """
    Add a file to the status.

    Parameters
    ----------
    file : dict
        File to add.

    Returns
    -------
    None
    """

    # Add the file to the list
    if self.files is None:
        self.files = []

    # Remove the file info if it already exists
    self.files = [f for f in self.files if f["path"] != file["path"]]

    # Add the new file
    self.files.append(file)

ModelStatusModel

Bases: ModelStatus

Status class for model entities.

Source code in digitalhub_ml/entities/models/status.py 
54
55
56
57
class ModelStatusModel(ModelStatus):
    """
    Status class for model entities.
    """