From 33ea700f6529face405c98322d369708322f43bd Mon Sep 17 00:00:00 2001 From: Christopher Harris Date: Wed, 13 Mar 2024 17:39:44 +0000 Subject: [PATCH] make flake8 happy --- examples/doca/run.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/examples/doca/run.py b/examples/doca/run.py index 6a38c83fe1..916f78650c 100644 --- a/examples/doca/run.py +++ b/examples/doca/run.py @@ -119,12 +119,13 @@ def run_pipeline(pipeline_batch_size, pipeline.add_stage(DeserializeStage(config)) pipeline.add_stage(MonitorStage(config, description="Deserialize rate", unit='pkts')) + hashfile = '/workspace/models/training-tuning-scripts/sid-models/resources/bert-base-uncased-hash.txt' + # add preprocessing stage pipeline.add_stage( PreprocessNLPStage( config, - vocab_hash_file= - '/workspace/models/training-tuning-scripts/sid-models/resources/bert-base-uncased-hash.txt', + vocab_hash_file=hashfile, do_lower_case=True, truncation=True, add_special_tokens=False,