FilesExpand file tree

 main

/

hardware.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

68 lines (48 loc) · 1.43 KB

 main

/

hardware.py

Copy path

Top

File metadata and controls

• Code
• Blame

68 lines (48 loc) · 1.43 KB

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

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

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

from typing import TYPE_CHECKING, Any, Dict, List

from typing_extensions import deprecated

from replicate.resource import Namespace, Resource

if TYPE_CHECKING:

pass

class Hardware(Resource):

"""

Hardware for running a model on Replicate.

"""

sku: str

"""

The SKU of the hardware.

"""

name: str

"""

The name of the hardware.

"""

@property

@deprecated("Use `sku` instead of `id`")

def id(self) -> str:

"""

DEPRECATED: Use `sku` instead.

"""

return self.sku

class HardwareNamespace(Namespace):

"""

Namespace for operations related to hardware.

"""

def list(self) -> List[Hardware]:

"""

List all hardware available for you to run models on Replicate.

Returns:

List[Hardware]: A list of hardware.

"""

resp = self._client._request("GET", "/v1/hardware")

obj = resp.json()

return [_json_to_hardware(entry) for entry in obj]

async def async_list(self) -> List[Hardware]:

"""

List all hardware available for you to run models on Replicate.

Returns:

List[Hardware]: A list of hardware.

"""

resp = await self._client._async_request("GET", "/v1/hardware")

obj = resp.json()

return [_json_to_hardware(entry) for entry in obj]

def _json_to_hardware(json: Dict[str, Any]) -> Hardware:

return Hardware(**json)