Skip to content

Commit

Permalink
Revert "Update summit/strategies/base.py (#257)"
Browse files Browse the repository at this point in the history
This reverts commit 820d9c6.
  • Loading branch information
marcosfelt committed Jun 21, 2023
1 parent 820d9c6 commit 7aea576
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions summit/strategies/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@ def transform_inputs_outputs(self, ds: DataSet, **kwargs):
self.input_means, self.input_stds = {}, {}
self.output_means, self.output_stds = {}, {}
self.encoders = {}
for variable in self.transform_domain.input_variables:
for variable in self.domain.input_variables:
if (
isinstance(variable, CategoricalVariable)
and categorical_method == "descriptors"
Expand Down Expand Up @@ -262,7 +262,7 @@ def un_transform(self, ds, **kwargs):

# Determine input and output columns in dataset
new_ds = ds.copy()
for i, variable in enumerate(self.transform_domain.input_variables):
for i, variable in enumerate(self.domain.input_variables):
# Categorical variables with descriptors
if (
isinstance(variable, CategoricalVariable)
Expand Down Expand Up @@ -1200,4 +1200,4 @@ def _design_distances(design_point, candidate_matrix):
squared = np.power(diff, 2)
summed = np.sum(squared, axis=1)
root_square = np.sqrt(summed)
return root_square
return root_square

0 comments on commit 7aea576

Please sign in to comment.