Skip to content

Commit 23fa2bb

Browse files
committed
refactor: simplify anchor_type initialization in AnchorBFSPartitioner
1 parent b21f4ce commit 23fa2bb

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

graphgen/models/partitioner/anchor_bfs_partitioner.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import random
22
from collections import deque
3-
from typing import Any, Iterable, List, Literal, Set, Tuple, Union
3+
from typing import Any, Iterable, List, Set, Tuple
44

55
from graphgen.bases import BaseGraphStorage
66
from graphgen.bases.datatypes import Community
@@ -22,10 +22,7 @@ class AnchorBFSPartitioner(BFSPartitioner):
2222

2323
def __init__(
2424
self,
25-
anchor_type: Union[
26-
Literal["image", "dna", "rna", "protein"],
27-
List[Literal["dna", "rna", "protein"]],
28-
] = "image",
25+
anchor_type: list = ["image"],
2926
anchor_ids: Set[str] | None = None,
3027
) -> None:
3128
super().__init__()

0 commit comments

Comments
 (0)