Decompose split into split and gather

ucbepic · Aug 22, 2024 · 0871456 · 0871456
1 parent de0c22f
commit 0871456
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/motion/operations/gather.py b/motion/operations/gather.py
@@ -1,4 +1,5 @@
 from typing import Dict, List, Tuple
+
 from motion.operations.base import BaseOperation
 
 

diff --git a/motion/operations/resolve.py b/motion/operations/resolve.py
@@ -7,7 +7,7 @@
 
 import jinja2
 from jinja2 import Template
-from litellm import completion_cost
+from litellm import completion_cost, embedding
 from sklearn.metrics.pairwise import cosine_similarity
 
 from motion.operations.base import BaseOperation
@@ -18,7 +18,6 @@
     rich_as_completed,
     validate_output,
 )
-from litellm import embedding
 
 
 def compare_pair(

diff --git a/motion/operations/split.py b/motion/operations/split.py
@@ -2,6 +2,7 @@
 from typing import Dict, List, Tuple
 
 import tiktoken
+
 from motion.operations.base import BaseOperation
Original file line number	Diff line number	Diff line change
		@@ -1,4 +1,5 @@
		from typing import Dict, List, Tuple

		from motion.operations.base import BaseOperation


Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2,6 +2,7 @@
		from typing import Dict, List, Tuple

		import tiktoken

		from motion.operations.base import BaseOperation


Expand Down