Method
__init__
(
self,
num_v: int,
e_list_v2e: Optional[Union[List[int], List[List[int]]]] = None,
e_list_e2v: Optional[Union[List[int], List[List[int]]]] = None,
w_list_v2e: Optional[Union[List[float], List[List[float]]]] = None,
w_list_e2v: Optional[Union[List[float], List[List[float]]]] = None,
e_weight: Optional[Union[float, List[float]]] = None,
v_weight: Optional[List[float]] = None,
device: torch.device = torch.device("cpu"),
)
Source from the content-addressed store, hash-verified
| 53 | """ |
| 54 | |
| 55 | def __init__( |
| 56 | self, |
| 57 | num_v: int, |
| 58 | e_list_v2e: Optional[Union[List[int], List[List[int]]]] = None, |
| 59 | e_list_e2v: Optional[Union[List[int], List[List[int]]]] = None, |
| 60 | w_list_v2e: Optional[Union[List[float], List[List[float]]]] = None, |
| 61 | w_list_e2v: Optional[Union[List[float], List[List[float]]]] = None, |
| 62 | e_weight: Optional[Union[float, List[float]]] = None, |
| 63 | v_weight: Optional[List[float]] = None, |
| 64 | device: torch.device = torch.device("cpu"), |
| 65 | ): |
| 66 | assert ( |
| 67 | isinstance(num_v, int) and num_v > 0 |
| 68 | ), "num_v should be a positive integer" |
| 69 | self.clear() |
| 70 | self._num_v = num_v |
| 71 | self.device = device |
| 72 | |
| 73 | @abc.abstractmethod |
| 74 | def __repr__(self) -> str: |
Callers
nothing calls this directly
Tested by
no test coverage detected