Skip to content

Commit a074531

Browse files
committed
[fix] Change int to np.int32 for the ndarray dtype specification
1 parent 224aa44 commit a074531

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

autoPyTorch/pipeline/components/setup/network_embedding/base_network_embedding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def _get_args(self, X: Dict[str, Any]) -> Tuple[int, np.ndarray]:
4444
num_numerical_columns = numerical_column_transformer.transform(
4545
X_train[:, X['dataset_properties']['numerical_columns']]).shape[1]
4646
num_input_features = np.zeros((num_numerical_columns + len(X['dataset_properties']['categorical_columns'])),
47-
dtype=int)
47+
dtype=np.int32)
4848
categories = X['dataset_properties']['categories']
4949

5050
for i, category in enumerate(categories):

autoPyTorch/pipeline/create_searchspace_util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def get_match_array(
4747
matches_dimensions = [len(choices) for choices in node_i_choices]
4848
# Start by allowing every combination of nodes. Go through all
4949
# combinations/pipelines and erase the illegal ones
50-
matches = np.ones(matches_dimensions, dtype=int)
50+
matches = np.ones(matches_dimensions, dtype=np.int32)
5151

5252
# TODO: Check if we need this, like are there combinations from the
5353
# pipeline we should dynamically avoid?

0 commit comments

Comments
 (0)