linting

owkin · Aug 12, 2024 · f6323fc · f6323fc
1 parent 3bee459
commit f6323fc
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/fedeca/tests/strategies/test_km.py b/fedeca/tests/strategies/test_km.py
@@ -5,7 +5,6 @@
 
 import git
 import numpy as np
-import pandas as pd
 import torch
 from lifelines import KaplanMeierFitter as KMF
 from substrafl.dependency import Dependency
@@ -19,7 +18,7 @@
 from fedeca.strategies.fed_kaplan import FedKaplan
 from fedeca.tests.common import TestTempDir
 from fedeca.utils.data_utils import split_dataframe_across_clients
-from fedeca.utils.survival_utils import CoxData, compute_events_statistics, km_curve
+from fedeca.utils.survival_utils import CoxData
 
 
 class TestKM(TestTempDir):