remove unused code

judithabk6 · Feb 11, 2025 · 54a0a6c · 54a0a6c
1 parent 48e3224
commit 54a0a6c
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/med_bench/estimation/base.py b/src/med_bench/estimation/base.py
@@ -1,7 +1,6 @@
 from abc import ABCMeta, abstractmethod
 import numpy as np
 from sklearn import clone
-from sklearn.model_selection import GridSearchCV
 
 from med_bench.utils.decorators import fitted
 
@@ -142,7 +141,6 @@ def _fit_treatment_propensity_xm(self, t, m, x):
 
         return self
 
-    # TODO : Enable any sklearn object as classifier or regressor
     def _fit_binary_mediator_probability(self, t, m, x):
         """Fits the nuisance parameter for the density f(M=m|T, X)"""
         # estimate mediator densities