neuml/txtai
1
0
Fork 0
mirror of https://github.com/neuml/txtai.git synced 2026-10-12 02:19:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Keep all rows for native vstack merges with list inputs (#1320)

Browse Source
This commit is contained in:
Edson
2026-10-02 14:30:31 -04:00
committed by GitHub
parent 00622f2cf1
commit 98c965cc3b
2 changed files with 19 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/python/testworkflow.py
+11
View File
@@ -391,6 +391,17 @@ class TestWorkflow(unittest.TestCase):
results = list(workflow(np.array([2, 4, 6])))
self.assertTrue(np.array_equal(np.array(results), np.array([[4, 16, 36], [8, 64, 216]])))
def testNumpyListWorkflow(self):
"""
Test a vstack merge of numpy and torch action outputs with list inputs
"""
# pylint: disable=E1101
for function in [lambda x, y: np.power(np.array(x), y), lambda x, y: torch.pow(torch.tensor(x), y)]:
task = Task([lambda x, f=function: f(x, 2), lambda x, f=function: f(x, 3)], merge="vstack")
results = list(Workflow([task])([2, 4]))
self.assertEqual([int(x) for x in results], [4, 8, 16, 64])
def testRetrieveWorkflow(self):
"""
Test a retrieve task