nnsight icon indicating copy to clipboard operation
nnsight copied to clipboard

Iterator context optionally returns Iterator object

Open AdamBelfki3 opened this issue 1 year ago • 0 comments

Accessing the Iterator object when creating an Iterator context has become optional.

with model.session() as session:
    l  = session.apply(list).save()
    with session.iter([0, 1, 2]) as item:
        l.append(item)

print("List: ", l)
"List: [0, 1, 2]"

If you wish to use the Iterator, simply set return_context=True.

with model.session() as session:
    l  = session.apply(list).save()
    with session.iter([0, 1, 2], return_context=True) as (item, iterator):
        l.append(item)
        iterator.exit()

print("List: ", l)
"List: [0]"

AdamBelfki3 avatar Aug 21 '24 21:08 AdamBelfki3