Skip to content

sgnts.base.notorch_backend

TorchArray

A placeholder to indicate TorchArray is not available

Source code in src/sgnts/base/notorch_backend.py 
93
94
95
96
97
98
99
class TorchArray:
    """A placeholder to indicate TorchArray is not available"""

    def __new__(cls, *args, **kwargs):
        raise ImportError(
            "PyTorch is not installed. Install it with 'pip install sgn-ts[torch]'"
        )

TorchBackend

Bases: ArrayBackend


              flowchart TD
              sgnts.base.notorch_backend.TorchBackend[TorchBackend]
              sgnts.base.array_backend.ArrayBackend[ArrayBackend]

                              sgnts.base.array_backend.ArrayBackend --> sgnts.base.notorch_backend.TorchBackend
                


              click sgnts.base.notorch_backend.TorchBackend href "" "sgnts.base.notorch_backend.TorchBackend"
              click sgnts.base.array_backend.ArrayBackend href "" "sgnts.base.array_backend.ArrayBackend"

A fallback TorchBackend that raises informative errors when torch is not available

Source code in src/sgnts/base/notorch_backend.py 
 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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
class TorchBackend(ArrayBackend):
    """A fallback TorchBackend that raises informative errors when torch is not
    available"""

    DTYPE = None
    DEVICE = None

    def __init__(self, *args, **kwargs):
        raise ImportError(
            "PyTorch is not installed. Install it with 'pip install sgn-ts[torch]'"
        )

    @staticmethod
    def all(*args, **kwargs):
        raise ImportError(
            "PyTorch is not installed. Install it with 'pip install sgn-ts[torch]'"
        )

    @classmethod
    def arange(cls, stop: float, start: float = 0, step: float = 1) -> Array:
        raise ImportError(
            "PyTorch is not installed. Install it with 'pip install sgn-ts[torch]'"
        )

    @staticmethod
    def cat(data: Sequence[Array], axis: int) -> Array:
        raise ImportError(
            "PyTorch is not installed. Install it with 'pip install sgn-ts[torch]'"
        )

    @classmethod
    def full(cls, shape: Tuple[int, ...], fill_value: Any) -> Array:
        raise ImportError(
            "PyTorch is not installed. Install it with 'pip install sgn-ts[torch]'"
        )

    @staticmethod
    def matmul(a: Array, b: Array) -> Array:
        raise ImportError(
            "PyTorch is not installed. Install it with 'pip install sgn-ts[torch]'"
        )

    @classmethod
    def ones(cls, shape: Tuple[int, ...]) -> Array:
        raise ImportError(
            "PyTorch is not installed. Install it with 'pip install sgn-ts[torch]'"
        )

    @staticmethod
    def pad(data: Array, pad_samples: tuple[int, int]) -> Array:
        raise ImportError(
            "PyTorch is not installed. Install it with 'pip install sgn-ts[torch]'"
        )

    @classmethod
    def set_device(cls, device: Union[str, Any]) -> None:
        raise ImportError(
            "PyTorch is not installed. Install it with 'pip install sgn-ts[torch]'"
        )

    @classmethod
    def set_dtype(cls, dtype: Any) -> None:
        raise ImportError(
            "PyTorch is not installed. Install it with 'pip install sgn-ts[torch]'"
        )

    @staticmethod
    def stack(data: Sequence[Array], axis: int = 0) -> Array:
        raise ImportError(
            "PyTorch is not installed. Install it with 'pip install sgn-ts[torch]'"
        )

    @staticmethod
    def sum(a: Array, axis: Optional[Union[int, tuple[int, ...]]] = None) -> Array:
        raise ImportError(
            "PyTorch is not installed. Install it with 'pip install sgn-ts[torch]'"
        )

    @classmethod
    def zeros(cls, shape: Tuple[int, ...]) -> Array:
        raise ImportError(
            "PyTorch is not installed. Install it with 'pip install sgn-ts[torch]'"
        )