FilesExpand file tree

 main

/

base_model.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

26 lines (19 loc) · 624 Bytes

 main

/

base_model.py

Copy path

Top

File metadata and controls

• Code
• Blame

26 lines (19 loc) · 624 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

from typing import TYPE_CHECKING

if TYPE_CHECKING:

from replicate.client import Client

from replicate.collection import Collection

import pydantic

class BaseModel(pydantic.BaseModel):

"""

A base class for representing a single object on the server.

"""

id: str

_client: "Client" = pydantic.PrivateAttr()

_collection: "Collection" = pydantic.PrivateAttr()

def reload(self) -> None:

"""

Load this object from the server again.

"""

new_model = self._collection.get(self.id)

for k, v in new_model.dict().items():

setattr(self, k, v)