@@ -25,9 +25,9 @@ from ..cpp.types.type_id cimport (type_id_t, uninitialized_id,
2525 type_id)
2626from ..cpp.types.datashape_formatter cimport format_datashape as dynd_format_datashape
2727# from ..cpp.types.categorical_type cimport dynd_make_categorical_type
28- from ..cpp.types.base_fixed_dim_type cimport dynd_make_fixed_dim_kind_type
2928from ..cpp.types.tuple_type cimport tuple_type
3029from ..cpp.types.struct_type cimport struct_type
30+ from ..cpp.types.base_fixed_dim_type cimport base_fixed_dim_type
3131from ..cpp.types.var_dim_type cimport var_dim_type as _var_dim_type
3232from ..cpp.types.callable_type cimport callable_type as _callable_type
3333from ..cpp.types.string_type cimport string_type
@@ -552,7 +552,7 @@ def make_struct(field_types, field_names):
552552 result.v = dynd_make_struct_type(field_types, field_names)
553553 return result
554554
555- def make_fixed_dim_kind (element_tp , ndim = None ):
555+ def make_fixed_dim_kind (element_tp ):
556556 """
557557 ndt.make_fixed_dim_kind(element_tp, ndim=1)
558558 Constructs an array dynd type with one or more symbolic fixed
@@ -574,10 +574,7 @@ def make_fixed_dim_kind(element_tp, ndim=None):
574574 ndt.type("Fixed * Fixed * Fixed * int32")
575575 """
576576 cdef type result = type ()
577- if (ndim is None ):
578- result.v = dynd_make_fixed_dim_kind_type(type (element_tp).v)
579- else :
580- result.v = dynd_make_fixed_dim_kind_type(type (element_tp).v, int (ndim))
577+ result.v = make_type[base_fixed_dim_type](type (element_tp).v)
581578 return result
582579
583580
0 commit comments