From 382c0a3b4565bf32665a3d043ad8be80706c0859 Mon Sep 17 00:00:00 2001 From: Julio Date: Wed, 1 Nov 2023 22:08:03 -0400 Subject: [PATCH] update file to pass lint action --- .../t4r_paper_repro/transf_exp_main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/t4rec_paper_experiments/t4r_paper_repro/transf_exp_main.py b/examples/t4rec_paper_experiments/t4r_paper_repro/transf_exp_main.py index c7ae93c6c..3cd0efedf 100644 --- a/examples/t4rec_paper_experiments/t4r_paper_repro/transf_exp_main.py +++ b/examples/t4rec_paper_experiments/t4r_paper_repro/transf_exp_main.py @@ -22,6 +22,7 @@ import numpy as np import pandas as pd import torch +import transformers from exp_outputs import ( config_dllogger, creates_output_dir, @@ -32,15 +33,14 @@ from merlin.io import Dataset from merlin.schema import Tags from transf_exp_args import DataArguments, ModelArguments, TrainingArguments +from transformers import HfArgumentParser, set_seed +from transformers.trainer_utils import is_main_process -import transformers import transformers4rec.torch as t4r from merlin_standard_lib import Schema -from transformers import HfArgumentParser, set_seed from transformers4rec.torch import Trainer from transformers4rec.torch.utils.data_utils import MerlinDataLoader from transformers4rec.torch.utils.examples_utils import wipe_memory -from transformers.trainer_utils import is_main_process logger = logging.getLogger(__name__)