Skip to content

Commit a9082a3

Browse files
authored
changed "train" to "fit" in the super call (#185)
Because of the name change from of the fitting method from "train" to "fit", there are some inconsistencies. Changed the super call, is it would break there
1 parent b3ac1b0 commit a9082a3

File tree

1 file changed

+1
-1
lines changed
  • src/ydata_synthetic/synthesizers/regular/wgangp

1 file changed

+1
-1
lines changed

src/ydata_synthetic/synthesizers/regular/wgangp/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ def fit(self, data, train_arguments: TrainParameters, num_cols: List[str], cat_c
142142
num_cols: List of columns of the data object to be handled as numerical
143143
cat_cols: List of columns of the data object to be handled as categorical
144144
"""
145-
super().train(data, num_cols, cat_cols)
145+
super().fit(data, num_cols, cat_cols)
146146

147147
processed_data = self.processor.transform(data)
148148
self.data_dim = processed_data.shape[1]

0 commit comments

Comments
 (0)