From 69fbd57808530417121030ea8aff5b1bae654c68 Mon Sep 17 00:00:00 2001 From: Oliver Holworthy Date: Mon, 5 Jun 2023 21:39:01 +0100 Subject: [PATCH] Use tmpdir for Categorify out_path in test_tf4rec (#1832) Co-authored-by: Adam Laiacano <108741458+nv-alaiacano@users.noreply.github.com> --- tests/unit/test_tf4rec.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/test_tf4rec.py b/tests/unit/test_tf4rec.py index 46dd66054a..737515e928 100644 --- a/tests/unit/test_tf4rec.py +++ b/tests/unit/test_tf4rec.py @@ -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), @@ -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) )