Skip to content

Commit

Permalink
update constant name
Browse files Browse the repository at this point in the history
  • Loading branch information
rjzamora committed Jul 30, 2024
1 parent 9bd271b commit 3886cd6
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions nvtabular/ops/categorify.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@
PAD_OFFSET = 0
NULL_OFFSET = 1
OOV_OFFSET = 2
PA_GE_14 = Version(pa.__version__) >= Version("14.0")
PYARROW_GE_14 = Version(pa.__version__) >= Version("14.0")


class Categorify(StatOperator):
Expand Down Expand Up @@ -909,7 +909,7 @@ def _general_concat(
):
# Concatenate DataFrame or pa.Table objects
if isinstance(frames[0], pa.Table):
if PA_GE_14:
if PYARROW_GE_14:
df = pa.concat_tables(frames, promote_options="default")
else:
df = pa.concat_tables(frames, promote=True)
Expand Down

0 comments on commit 3886cd6

Please sign in to comment.