fix(sagemaker.py): fix token iterator default flag

This commit is contained in:
Krrish Dholakia 2024-02-13 21:41:09 -08:00
parent 3b1cef3e96
commit 3ef391800a

View file

@ -30,7 +30,7 @@ import json
class TokenIterator:
def __init__(self, stream, acompletion: bool):
def __init__(self, stream, acompletion: bool = False):
if acompletion == False:
self.byte_iterator = iter(stream)
elif acompletion == True: