fixed missing import

ndif-team · Dec 18, 2024 · f4641a7 · f4641a7
1 parent 407b8f7
commit f4641a7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/nnsight/intervention/base.py b/src/nnsight/intervention/base.py
@@ -11,7 +11,8 @@
 from .contexts import EditingTracer, InterleavingTracer, Session
 from .envoy import Envoy
 from .graph import (InterventionGraph, InterventionNode, InterventionProxy,
-                    InterventionProxyType, Proxy)
+                    InterventionProxyType)
+from .graph.proxy import Proxy
 from .interleaver import Interleaver
 from .. import CONFIG