[air] fix xgboost_benchmark script by passing in args (#27146)

This commit is contained in:
matthewdeng 2022-07-27 19:08:15 -07:00 committed by GitHub
parent db26c779a0
commit 0319dcd889
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -58,7 +58,7 @@ def run_and_time_it(f):
self._exception = self._pconn.recv()
return self._exception
p = MyProcess(target=f, *args, **kwargs)
p = MyProcess(target=f, args=args, kwargs=kwargs)
start = time.monotonic()
p.start()
p.join()