Use simple print in tracker print function. (#6609)

This commit is contained in:
Jiaming Yuan
2021-01-21 21:15:43 +08:00
committed by GitHub
parent 26982f9fce
commit 7bc56fa0ed
3 changed files with 8 additions and 6 deletions

View File

@@ -867,7 +867,9 @@ class TestWithDask:
test = "--gtest_filter=Quantile." + name
def runit(worker_addr: str, rabit_args: List[bytes]) -> subprocess.CompletedProcess:
def runit(
worker_addr: str, rabit_args: List[bytes]
) -> subprocess.CompletedProcess:
port_env = ''
# setup environment for running the c++ part.
for arg in rabit_args:
@@ -965,7 +967,9 @@ class TestWithDask:
with tempfile.TemporaryDirectory() as tmpdir:
path = os.path.join(tmpdir, 'log')
def sqr(labels: np.ndarray, predts: np.ndarray) -> Tuple[np.ndarray, np.ndarray]:
def sqr(
labels: np.ndarray, predts: np.ndarray
) -> Tuple[np.ndarray, np.ndarray]:
with open(path, 'a') as fd:
print('Running sqr', file=fd)
grad = predts - labels