Skip to content
Snippets Groups Projects
Commit 78230b72 authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Some missing 'NumpyType' wrappers in nbody

parent 0c23b682
No related branches found
No related tags found
Loading
...@@ -59,13 +59,13 @@ class ReductionOperation(object): ...@@ -59,13 +59,13 @@ class ReductionOperation(object):
@staticmethod @staticmethod
def parse_result_type(target, op_type): def parse_result_type(target, op_type):
try: try:
return np.dtype(op_type) return NumpyType(np.dtype(op_type))
except TypeError: except TypeError:
pass pass
if op_type.startswith("vec_"): if op_type.startswith("vec_"):
try: try:
return target.get_or_register_dtype(op_type[4:]) return NumpyType(target.get_or_register_dtype(op_type[4:]))
except AttributeError: except AttributeError:
pass pass
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment