We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d3edeed commit ca240b5Copy full SHA for ca240b5
1 file changed
src/optimagic/parameters/tree_registry.py
@@ -1,6 +1,5 @@
1
"""Wrapper around optree to tailor it to optimagic."""
2
3
-import itertools
4
from functools import partial
5
from itertools import product
6
@@ -165,7 +164,7 @@ def _index_element_to_string(element):
165
164
166
def _array_element_names(arr):
167
dim_names = [map(str, range(n)) for n in arr.shape]
168
- names = list(map("_".join, itertools.product(*dim_names)))
+ names = list(map("_".join, product(*dim_names)))
169
return names
170
171
0 commit comments