Filipstrozik
commited on
Commit
路
8e9893b
1
Parent(s):
afc2161
Remove unused Self type hint from split method in RegressorPrediction class
Browse files
ellipse_rcnn/core/ellipse_roi_head.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
from typing import Dict, List, Tuple, Optional, TypedDict, NamedTuple
|
2 |
|
3 |
import torch
|
4 |
from torch import nn
|
@@ -69,7 +69,7 @@ class RegressorPrediction(NamedTuple):
|
|
69 |
def dtype(self) -> torch.dtype:
|
70 |
return self.d_a.dtype
|
71 |
|
72 |
-
def split(self, split_size: list[int] | int, dim: int = 0)
|
73 |
return [
|
74 |
RegressorPrediction(*tensors)
|
75 |
for tensors in zip(
|
|
|
1 |
+
from typing import Dict, List, Tuple, Optional, TypedDict, NamedTuple
|
2 |
|
3 |
import torch
|
4 |
from torch import nn
|
|
|
69 |
def dtype(self) -> torch.dtype:
|
70 |
return self.d_a.dtype
|
71 |
|
72 |
+
def split(self, split_size: list[int] | int, dim: int = 0):
|
73 |
return [
|
74 |
RegressorPrediction(*tensors)
|
75 |
for tensors in zip(
|