Neither export_pairs_to_csv nor export_to_df works on EntityMatching object for clean-clean ER

[zmbc]

Describe the bug
export_pairs_to_csv and export_to_df both fail with KeyError on an EntityMatching object for clean-clean ER.

To Reproduce

import pandas as pd
from pyjedai.datamodel import Data

d1 = pd.read_csv("https://raw.githubusercontent.com/AI-team-UoA/pyJedAI/main/data/ccer/D2/abt.csv", sep='|', engine='python', na_filter=False)
d2 = pd.read_csv("https://raw.githubusercontent.com/AI-team-UoA/pyJedAI/main/data/ccer/D2/buy.csv", sep='|', engine='python', na_filter=False)
gt = pd.read_csv("https://raw.githubusercontent.com/AI-team-UoA/pyJedAI/main/data/ccer/D2/gt.csv", sep='|', engine='python')

data = Data(dataset_1=d1,
            id_column_name_1='id',
            dataset_2=d2,
            id_column_name_2='id',
            ground_truth=gt)

from pyjedai.block_building import StandardBlocking

bb = StandardBlocking()
blocks = bb.build_blocks(data, attributes_1=['name'], attributes_2=['name'])

from pyjedai.matching import EntityMatching

em = EntityMatching(
    metric='cosine',
    tokenizer='char_tokenizer',
    vectorizer='tfidf',
    qgram=3,
    similarity_threshold=0.0
)

pairs_graph = em.predict(blocks, data, tqdm_disable=True)

em.export_pairs_to_csv('foo.csv') # Fails with "KeyError: 0"
em.export_to_df(pairs_graph) # Fails with "KeyError: 1094"

Expected behavior

A CSV file or dataframe of pairs to be generated.

[Nikoletos-K]

Hello there!

Thanks for the detailed report. Yeah there's a mistake. I have fixed it, in dev-stage.
These changes will be available in the next few days (if not today) in a new release.

Also, fyi, export_to_csv will be removed and you should export_to_df and then call pandas to_csv().

Cheers,
Konstantinos

[Nikoletos-K]

@zmbc Check new release 0.1.9. If it's okay ping me to close this issue.

[zmbc]

Thank you so much for the prompt response and fix! I'll test it out.

[zmbc]

It works now, thank you 😃