Skip to content
This repository was archived by the owner on Feb 2, 2024. It is now read-only.

Commit 65fc103

Browse files
author
Ehsan Totoni
committed
fix aliasing for Series with index (TODO: test)
1 parent 58bb4b6 commit 65fc103

1 file changed

Lines changed: 9 additions & 1 deletion

File tree

hpat/hiframes/api.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -907,8 +907,16 @@ def alias_ext_dummy_func(lhs_name, args, alias_map, arg_aliases):
907907
assert len(args) >= 1
908908
numba.ir_utils._add_alias(lhs_name, args[0].name, alias_map, arg_aliases)
909909

910+
911+
def alias_ext_init_series(lhs_name, args, alias_map, arg_aliases):
912+
assert len(args) >= 1
913+
numba.ir_utils._add_alias(lhs_name, args[0].name, alias_map, arg_aliases)
914+
if len(args) > 1: # has index
915+
numba.ir_utils._add_alias(lhs_name, args[1].name, alias_map, arg_aliases)
916+
917+
910918
if hasattr(numba.ir_utils, 'alias_func_extensions'):
911-
numba.ir_utils.alias_func_extensions[('init_series', 'hpat.hiframes.api')] = alias_ext_dummy_func
919+
numba.ir_utils.alias_func_extensions[('init_series', 'hpat.hiframes.api')] = alias_ext_init_series
912920
numba.ir_utils.alias_func_extensions[('get_series_data', 'hpat.hiframes.api')] = alias_ext_dummy_func
913921
numba.ir_utils.alias_func_extensions[('get_series_index', 'hpat.hiframes.api')] = alias_ext_dummy_func
914922
numba.ir_utils.alias_func_extensions[('init_datetime_index', 'hpat.hiframes.api')] = alias_ext_dummy_func

0 commit comments

Comments
 (0)