Skip to content

Commit a28f91e

Browse files
committed
chore: self reference without quotations marks
1 parent 372dcd7 commit a28f91e

File tree

2 files changed

+53
-51
lines changed

2 files changed

+53
-51
lines changed

lantern/numpy.py

Lines changed: 25 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from __future__ import annotations
12
import numpy as np
23
import torch
34

@@ -19,7 +20,7 @@ def validate(cls, data) -> np.ndarray:
1920
return np.array(data)
2021

2122
@classmethod
22-
def ndim(cls, ndim) -> "Numpy":
23+
def ndim(cls, ndim) -> Numpy:
2324
class InheritNumpy(cls):
2425
@classmethod
2526
def validate(cls, data):
@@ -31,7 +32,7 @@ def validate(cls, data):
3132
return InheritNumpy
3233

3334
@classmethod
34-
def dims(cls, dims) -> "Numpy":
35+
def dims(cls, dims) -> Numpy:
3536
class InheritNumpy(cls):
3637
@classmethod
3738
def validate(cls, data):
@@ -45,7 +46,7 @@ def validate(cls, data):
4546
return InheritNumpy
4647

4748
@classmethod
48-
def shape(cls, *sizes) -> "Numpy":
49+
def shape(cls, *sizes) -> Numpy:
4950
class InheritNumpy(cls):
5051
@classmethod
5152
def validate(cls, data):
@@ -58,7 +59,7 @@ def validate(cls, data):
5859
return InheritNumpy
5960

6061
@classmethod
61-
def between(cls, ge, le) -> "Numpy":
62+
def between(cls, ge, le) -> Numpy:
6263
class InheritNumpy(cls):
6364
@classmethod
6465
def validate(cls, data):
@@ -78,7 +79,7 @@ def validate(cls, data):
7879
return InheritNumpy
7980

8081
@classmethod
81-
def ge(cls, ge) -> "Numpy":
82+
def ge(cls, ge) -> Numpy:
8283
class InheritTensor(cls):
8384
@classmethod
8485
def validate(cls, data):
@@ -91,7 +92,7 @@ def validate(cls, data):
9192
return InheritTensor
9293

9394
@classmethod
94-
def le(cls, le) -> "Numpy":
95+
def le(cls, le) -> Numpy:
9596
class InheritTensor(cls):
9697
@classmethod
9798
def validate(cls, data):
@@ -106,7 +107,7 @@ def validate(cls, data):
106107
return InheritTensor
107108

108109
@classmethod
109-
def gt(cls, gt) -> "Numpy":
110+
def gt(cls, gt) -> Numpy:
110111
class InheritTensor(cls):
111112
@classmethod
112113
def validate(cls, data):
@@ -118,7 +119,7 @@ def validate(cls, data):
118119
return InheritTensor
119120

120121
@classmethod
121-
def lt(cls, lt) -> "Numpy":
122+
def lt(cls, lt) -> Numpy:
122123
class InheritTensor(cls):
123124
@classmethod
124125
def validate(cls, data):
@@ -131,7 +132,7 @@ def validate(cls, data):
131132
return InheritTensor
132133

133134
@classmethod
134-
def ne(cls, ne) -> "Numpy":
135+
def ne(cls, ne) -> Numpy:
135136
class InheritTensor(cls):
136137
@classmethod
137138
def validate(cls, data):
@@ -144,7 +145,7 @@ def validate(cls, data):
144145
return InheritTensor
145146

146147
@classmethod
147-
def dtype(cls, dtype) -> "Numpy":
148+
def dtype(cls, dtype) -> Numpy:
148149
class InheritNumpy(cls):
149150
@classmethod
150151
def validate(cls, data):
@@ -157,63 +158,63 @@ def validate(cls, data):
157158
return InheritNumpy
158159

159160
@classmethod
160-
def float(cls) -> "Numpy":
161+
def float(cls) -> Numpy:
161162
return cls.dtype(np.float32)
162163

163164
@classmethod
164-
def float32(cls) -> "Numpy":
165+
def float32(cls) -> Numpy:
165166
return cls.dtype(np.float32)
166167

167168
@classmethod
168-
def half(cls) -> "Numpy":
169+
def half(cls) -> Numpy:
169170
return cls.dtype(np.float16)
170171

171172
@classmethod
172173
def float16(cls):
173174
return cls.dtype(np.float16)
174175

175176
@classmethod
176-
def double(cls) -> "Numpy":
177+
def double(cls) -> Numpy:
177178
return cls.dtype(np.float64)
178179

179180
@classmethod
180-
def float64(cls) -> "Numpy":
181+
def float64(cls) -> Numpy:
181182
return cls.dtype(np.float64)
182183

183184
@classmethod
184-
def int(cls) -> "Numpy":
185+
def int(cls) -> Numpy:
185186
return cls.dtype(np.int32)
186187

187188
@classmethod
188-
def int32(cls) -> "Numpy":
189+
def int32(cls) -> Numpy:
189190
return cls.dtype(np.int32)
190191

191192
@classmethod
192-
def long(cls) -> "Numpy":
193+
def long(cls) -> Numpy:
193194
return cls.dtype(np.int64)
194195

195196
@classmethod
196-
def int64(cls) -> "Numpy":
197+
def int64(cls) -> Numpy:
197198
return cls.dtype(np.int64)
198199

199200
@classmethod
200-
def short(cls) -> "Numpy":
201+
def short(cls) -> Numpy:
201202
return cls.dtype(np.int16)
202203

203204
@classmethod
204-
def int16(cls) -> "Numpy":
205+
def int16(cls) -> Numpy:
205206
return cls.dtype(np.int16)
206207

207208
@classmethod
208-
def byte(cls) -> "Numpy":
209+
def byte(cls) -> Numpy:
209210
return cls.dtype(np.uint8)
210211

211212
@classmethod
212-
def uint8(cls) -> "Numpy":
213+
def uint8(cls) -> Numpy:
213214
return cls.dtype(np.uint8)
214215

215216
@classmethod
216-
def bool(cls) -> "Numpy":
217+
def bool(cls) -> Numpy:
217218
return cls.dtype(bool)
218219

219220

lantern/tensor.py

Lines changed: 28 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from __future__ import annotations
12
import numpy as np
23
import torch
34

@@ -19,7 +20,7 @@ def validate(cls, data) -> torch.Tensor:
1920
return torch.as_tensor(data)
2021

2122
@classmethod
22-
def ndim(cls, ndim) -> "Tensor":
23+
def ndim(cls, ndim) -> Tensor:
2324
class InheritTensor(cls):
2425
@classmethod
2526
def validate(cls, data):
@@ -31,7 +32,7 @@ def validate(cls, data):
3132
return InheritTensor
3233

3334
@classmethod
34-
def dims(cls, dims) -> "Tensor":
35+
def dims(cls, dims) -> Tensor:
3536
class InheritTensor(cls):
3637
@classmethod
3738
def validate(cls, data):
@@ -45,7 +46,7 @@ def validate(cls, data):
4546
return InheritTensor
4647

4748
@classmethod
48-
def shape(cls, *sizes) -> "Tensor":
49+
def shape(cls, *sizes) -> Tensor:
4950
class InheritTensor(cls):
5051
@classmethod
5152
def validate(cls, data):
@@ -58,7 +59,7 @@ def validate(cls, data):
5859
return InheritTensor
5960

6061
@classmethod
61-
def between(cls, ge, le) -> "Tensor":
62+
def between(cls, ge, le) -> Tensor:
6263
class InheritTensor(cls):
6364
@classmethod
6465
def validate(cls, data):
@@ -77,7 +78,7 @@ def validate(cls, data):
7778
return InheritTensor
7879

7980
@classmethod
80-
def ge(cls, ge) -> "Tensor":
81+
def ge(cls, ge) -> Tensor:
8182
class InheritTensor(cls):
8283
@classmethod
8384
def validate(cls, data):
@@ -90,7 +91,7 @@ def validate(cls, data):
9091
return InheritTensor
9192

9293
@classmethod
93-
def le(cls, le) -> "Tensor":
94+
def le(cls, le) -> Tensor:
9495
class InheritTensor(cls):
9596
@classmethod
9697
def validate(cls, data):
@@ -105,7 +106,7 @@ def validate(cls, data):
105106
return InheritTensor
106107

107108
@classmethod
108-
def gt(cls, gt) -> "Tensor":
109+
def gt(cls, gt) -> Tensor:
109110
class InheritTensor(cls):
110111
@classmethod
111112
def validate(cls, data):
@@ -117,7 +118,7 @@ def validate(cls, data):
117118
return InheritTensor
118119

119120
@classmethod
120-
def lt(cls, lt) -> "Tensor":
121+
def lt(cls, lt) -> Tensor:
121122
class InheritTensor(cls):
122123
@classmethod
123124
def validate(cls, data):
@@ -130,7 +131,7 @@ def validate(cls, data):
130131
return InheritTensor
131132

132133
@classmethod
133-
def ne(cls, ne) -> "Tensor":
134+
def ne(cls, ne) -> Tensor:
134135
class InheritTensor(cls):
135136
@classmethod
136137
def validate(cls, data):
@@ -143,7 +144,7 @@ def validate(cls, data):
143144
return InheritTensor
144145

145146
@classmethod
146-
def device(cls, device) -> "Tensor":
147+
def device(cls, device) -> Tensor:
147148
class InheritTensor(cls):
148149
@classmethod
149150
def validate(cls, data):
@@ -152,15 +153,15 @@ def validate(cls, data):
152153
return InheritTensor
153154

154155
@classmethod
155-
def cpu(cls) -> "Tensor":
156+
def cpu(cls) -> Tensor:
156157
return cls.device(torch.device("cpu"))
157158

158159
@classmethod
159-
def cuda(cls) -> "Tensor":
160+
def cuda(cls) -> Tensor:
160161
return cls.device(torch.device("cuda"))
161162

162163
@classmethod
163-
def dtype(cls, dtype) -> "Tensor":
164+
def dtype(cls, dtype) -> Tensor:
164165
class InheritTensor(cls):
165166
@classmethod
166167
def validate(cls, data):
@@ -173,63 +174,63 @@ def validate(cls, data):
173174
return InheritTensor
174175

175176
@classmethod
176-
def float(cls) -> "Tensor":
177+
def float(cls) -> Tensor:
177178
return cls.dtype(torch.float32)
178179

179180
@classmethod
180-
def float32(cls) -> "Tensor":
181+
def float32(cls) -> Tensor:
181182
return cls.dtype(torch.float32)
182183

183184
@classmethod
184-
def half(cls) -> "Tensor":
185+
def half(cls) -> Tensor:
185186
return cls.dtype(torch.float16)
186187

187188
@classmethod
188189
def float16(cls):
189190
return cls.dtype(torch.float16)
190191

191192
@classmethod
192-
def double(cls) -> "Tensor":
193+
def double(cls) -> Tensor:
193194
return cls.dtype(torch.float64)
194195

195196
@classmethod
196-
def float64(cls) -> "Tensor":
197+
def float64(cls) -> Tensor:
197198
return cls.dtype(torch.float64)
198199

199200
@classmethod
200-
def int(cls) -> "Tensor":
201+
def int(cls) -> Tensor:
201202
return cls.dtype(torch.int32)
202203

203204
@classmethod
204-
def int32(cls) -> "Tensor":
205+
def int32(cls) -> Tensor:
205206
return cls.dtype(torch.int32)
206207

207208
@classmethod
208-
def long(cls) -> "Tensor":
209+
def long(cls) -> Tensor:
209210
return cls.dtype(torch.int64)
210211

211212
@classmethod
212-
def int64(cls) -> "Tensor":
213+
def int64(cls) -> Tensor:
213214
return cls.dtype(torch.int64)
214215

215216
@classmethod
216-
def short(cls) -> "Tensor":
217+
def short(cls) -> Tensor:
217218
return cls.dtype(torch.int16)
218219

219220
@classmethod
220-
def int16(cls) -> "Tensor":
221+
def int16(cls) -> Tensor:
221222
return cls.dtype(torch.int16)
222223

223224
@classmethod
224-
def byte(cls) -> "Tensor":
225+
def byte(cls) -> Tensor:
225226
return cls.dtype(torch.uint8)
226227

227228
@classmethod
228-
def uint8(cls) -> "Tensor":
229+
def uint8(cls) -> Tensor:
229230
return cls.dtype(torch.uint8)
230231

231232
@classmethod
232-
def bool(cls) -> "Tensor":
233+
def bool(cls) -> Tensor:
233234
return cls.dtype(torch.bool)
234235

235236

0 commit comments

Comments
 (0)