Skip to content

Commit

Permalink
add maxweight arg
Browse files Browse the repository at this point in the history
  • Loading branch information
maciejka committed Oct 23, 2024
1 parent 2170c4e commit 08d4a76
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions scripts/data/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -263,10 +263,17 @@ def main(start, blocks, step, mode, strategy):
choices=["sequential", "random"],
help="Processing strategy",
)

parser.add_argument(
"--maxweight", type=int, default=MAX_WEIGHT_LIMIT, help="Max weight limit"
)

parser.add_argument("--verbose", action="store_true", help="Verbose")

args = parser.parse_args()

MAX_WEIGHT_LIMIT=args.maxweight

file_handler = logging.FileHandler("client.errors.log")
file_handler.setLevel(logging.INFO)
file_handler.setFormatter(
Expand Down

0 comments on commit 08d4a76

Please sign in to comment.