diff --git a/src/defigraph/Edge.py b/src/defigraph/Edge.py index 2044ebe..1499e6e 100644 --- a/src/defigraph/Edge.py +++ b/src/defigraph/Edge.py @@ -2,12 +2,11 @@ from defigraph.Pool import Pool class Edge: - def __init__(self, u: Vertex, v: Vertex, weight: float=None): + def __init__(self, u: Vertex, v: Vertex, pool_address: str, weight: float=None): self.u = u self.v = v self.weight = weight - address = "" - self.pool = Pool(address=address,token0=u, token1=v) + self.pool = Pool(address=pool_address,token0=u, token1=v) def __repr__(self): if not self.weight: