Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion tests/sources/test_transformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

import pytest
from lxml import etree
from timdex_dataset_api.record import DatasetRecord
from timdex_dataset_api import DatasetRecord

import transmogrifier.models as timdex
from transmogrifier.exceptions import DeletedRecordEvent, SkippedRecordEvent
Expand Down
2 changes: 1 addition & 1 deletion transmogrifier/sources/transformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -383,7 +383,7 @@ def record_is_excluded(self, _source_record: dict[str, JSON] | Tag) -> bool:
def write_to_parquet_dataset(self, dataset_location: str) -> list:
"""Write output to TIMDEX dataset."""
timdex_dataset = TIMDEXDataset(location=dataset_location)
return timdex_dataset.write(records_iter=self)
return timdex_dataset.records.write(rows_iter=self)

@final
def get_valid_title(self, source_record: dict[str, JSON] | Tag) -> str:
Expand Down
Loading
Loading