fix minor bug

logix-project · Apr 23, 2024 · 28d16b6 · 28d16b6
1 parent 043d56d
commit 28d16b6
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/logix/analysis/influence_function.py b/logix/analysis/influence_function.py
@@ -18,7 +18,6 @@
     cross_dot_product,
     merge_influence_results,
 )
-from logix.statistic.utils import make_2d
 
 
 class InfluenceFunction:

diff --git a/logix/analysis/influence_function_utils.py b/logix/analysis/influence_function_utils.py
@@ -4,6 +4,7 @@
 from einops import einsum, reduce, rearrange
 
 from logix.state import LogIXState
+from logix.statistic.utils import make_2d
 from logix.utils import nested_dict