From c3c763a29b6fb9d8ebcac52df5d357548227e82a Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Thu, 14 May 2026 12:19:33 +0800 Subject: [PATCH] up --- reme2/component/file_graph/local_file_graph.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/reme2/component/file_graph/local_file_graph.py b/reme2/component/file_graph/local_file_graph.py index 12f63659..ddbf6341 100644 --- a/reme2/component/file_graph/local_file_graph.py +++ b/reme2/component/file_graph/local_file_graph.py @@ -20,13 +20,12 @@ Persistence: pickle to ``graph.pkl`` on close, restored on _start. from __future__ import annotations import pickle -from collections.abc import AsyncIterator from pathlib import Path from .base_file_graph import BaseFileGraph from ..component_registry import R from ...schema import FileLink, FileNode -import networkx as nx + @R.register("local") class LocalFileGraph(BaseFileGraph): @@ -41,6 +40,7 @@ class LocalFileGraph(BaseFileGraph): async def _start(self) -> None: await super()._start() + import networkx as nx self._graph = self._load_graph(nx) or nx.MultiDiGraph() self.logger.info( f"LocalFileGraph '{self.graph_name}' ready: "