在图形中查找顶点的邻居-修改后的bfs

Question

我是python的初学者，所以我将非常感谢您的帮助。 ：）

我需要编写一个函数，使用修改的bfs返回给定距离内给定顶点的所有邻居。我的代码在下面，但我不知道如何更正。

def bfs_neighbors(graph: List, first_vertex: int, distance: int) -> Set:
    visited, queue = set(), [first_vertex]
    for i in range(distance):
        while queue:
            vertex = queue.pop(0)
            if vertex not in queue:
                visited.add(vertex)
                if not visited:
                    queue.extend(graph[vertex])
        return visited