print

deepjavalibrary · Jul 6, 2023 · 3c233c5 · 3c233c5
1 parent 06bf60b
commit 3c233c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/engines/python/setup/djl_python/tests/test_scheduler.py b/engines/python/setup/djl_python/tests/test_scheduler.py
@@ -158,7 +158,7 @@ def test_sampling_scheduler(self):
 
         print('=====++++++++++==============')
         for i, ret in results.items():
-            print('\n{}:'.format(i), tokenizer.decode(ret))
+            print('\n{}:'.format(i), tokenizer.decode(ret[:30]))
 
         # assert tokenizer.decode(results[1][:30]) == "When your legs don't work like they used to before And I can't " \
         #                                             "sweep you off your feet, you're right, I'm done for the"