format errors are fixed

dice-group · Oct 25, 2024 · 64b0487 · 64b0487
1 parent 5612303
commit 64b0487
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dicee/read_preprocess_save_load_kg/util.py b/dicee/read_preprocess_save_load_kg/util.py
@@ -10,7 +10,7 @@
 import psutil
 import requests
 from typing import Tuple
-import pandas as pd
+
 def polars_dataframe_indexer(df_polars:polars.DataFrame, idx_entity:polars.DataFrame, idx_relation:polars.DataFrame)->polars.DataFrame:
     """
      Replaces 'subject', 'relation', and 'object' columns in the input Polars DataFrame with their corresponding index values

diff --git a/dicee/trainer/dice_trainer.py b/dicee/trainer/dice_trainer.py
@@ -4,7 +4,7 @@
 from dicee.models.base_model import BaseKGE
 from dicee.static_funcs import select_model
 from dicee.callbacks import ASWA, Eval, KronE, PrintCallback, AccumulateEpochLossCallback, Perturb
-from dicee.dataset_classes import construct_dataset, reload_dataset
+from dicee.dataset_classes import construct_dataset
 from .torch_trainer import TorchTrainer
 from .torch_trainer_ddp import TorchDDPTrainer
 from ..static_funcs import timeit