diff --git a/arekit/contrib/utils/serializer.py b/arekit/contrib/utils/serializer.py index 8c788ce0..4f6f08e8 100644 --- a/arekit/contrib/utils/serializer.py +++ b/arekit/contrib/utils/serializer.py @@ -56,5 +56,5 @@ def serialize(pipeline, doc_ids_iter, keep_labels, do_balance, sample_rows_provi samples_repo.balance() # Write repositories - samples_repo.write(writer=samples_writer, target=samples_target()) - opinions_repo.write(writer=opinions_writer, target=opinions_target()) + samples_repo.write(writer=samples_writer, target=samples_target) + opinions_repo.write(writer=opinions_writer, target=opinions_target)