Skip to content

Commit

Permalink
Use tmpdir for Categorify out_path in test_tf4rec
Browse files Browse the repository at this point in the history
  • Loading branch information
oliverholworthy committed Jun 5, 2023
1 parent bab6828 commit a87267a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/unit/test_tf4rec.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
NUM_ROWS = 10000


def test_tf4rec():
def test_tf4rec(tmpdir):
inputs = {
"user_session": np.random.randint(1, 10000, NUM_ROWS),
"product_id": np.random.randint(1, 51996, NUM_ROWS),
Expand All @@ -29,7 +29,7 @@ def test_tf4rec():

cat_feats = (
["user_session", "product_id", "category_id"]
>> nvt.ops.Categorify()
>> nvt.ops.Categorify(out_path=str(tmpdir))
>> nvt.ops.LambdaOp(lambda col: col + 1)
)

Expand Down

0 comments on commit a87267a

Please sign in to comment.