fixed missing import

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