Skip to content

Commit 85d9891

Browse files
committed
MOD: Add check for metadata ts_out in iteration
1 parent 6e10b81 commit 85d9891

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

databento/common/dbnstore.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -354,6 +354,7 @@ def __init__(self, data_source: DataSource) -> None:
354354

355355
def __iter__(self) -> Generator[DBNStruct, None, None]:
356356
reader = self.reader
357+
attach_ts_out = self.metadata.ts_out
357358
decoder = DbnDecoder()
358359
while True:
359360
raw = reader.read(DBNStore.DBN_READ_SIZE)
@@ -364,7 +365,7 @@ def __iter__(self) -> Generator[DBNStruct, None, None]:
364365
except ValueError:
365366
continue
366367
for record, ts_out in records:
367-
if not isinstance(record, Metadata):
368+
if attach_ts_out and not isinstance(record, Metadata):
368369
setattr(record, "ts_out", ts_out)
369370
yield record
370371
else:

0 commit comments

Comments
 (0)