Oops, missed a spot in simplification

alteryx · Apr 30, 2021 · bd9f6d0 · bd9f6d0
1 parent ed05271
commit bd9f6d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/evalml/pipelines/components/transformers/samplers/base_sampler.py b/evalml/pipelines/components/transformers/samplers/base_sampler.py
@@ -103,7 +103,7 @@ def fit(self, X, y):
         sampling_ratio = self.parameters['sampling_ratio']
         dic = make_balancing_dictionary(y_pd, sampling_ratio)
         sampler_params['sampling_strategy'] = dic
-        sampler = sampler_class(**sampler_params, random_state=self.random_seed)
+        sampler = self.sampler(**sampler_params, random_state=self.random_seed)
         self._component_obj = sampler
 
     def fit_transform(self, X, y):