Fix format errors

dice-group · Nov 21, 2023 · 1e66aa9 · 1e66aa9
1 parent f7e5ac2
commit 1e66aa9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dicee/analyse_experiments.py b/dicee/analyse_experiments.py
@@ -129,7 +129,7 @@ def analyse(args):
                 report = {i: report[i] for i in ['Runtime', 'NumParam']}
             with open(f'{full_path}/eval_report.json', 'r') as f:
                 eval_report = json.load(f)
-        except:
+        except FileNotFoundError:
             print("NOT found")
             continue
         config.update(eval_report)

diff --git a/dicee/trainer/dice_trainer.py b/dicee/trainer/dice_trainer.py
@@ -3,7 +3,7 @@
 from typing import Union
 from dicee.models.base_model import BaseKGE
 from dicee.static_funcs import select_model
-from dicee.callbacks import (ASWA, PPE, FPPE, Eval, KronE, PrintCallback, KGESaveCallback, AccumulateEpochLossCallback,
+from dicee.callbacks import (ASWA, PPE, FPPE, Eval, KronE, PrintCallback, AccumulateEpochLossCallback,
                              Perturb)
 from dicee.dataset_classes import construct_dataset, reload_dataset
 from .torch_trainer import TorchTrainer