diff --git a/flaml/automl/model.py b/flaml/automl/model.py index dcde0927ab..298f3cab80 100644 --- a/flaml/automl/model.py +++ b/flaml/automl/model.py @@ -2820,7 +2820,7 @@ def __init__(self): # Open a pair of null files self.null_fds = [os.open(os.devnull, os.O_RDWR) for x in range(2)] # Save the actual stdout (1) and stderr (2) file descriptors. - self.save_fds = (os.dup(1), os.dup(2)) + self.save_fds = [os.dup(1), os.dup(2)] def __enter__(self): # Assign the null pointers to stdout and stderr. @@ -2832,5 +2832,5 @@ def __exit__(self, *_): os.dup2(self.save_fds[0], 1) os.dup2(self.save_fds[1], 2) # Close the null files - os.close(self.null_fds[0]) - os.close(self.null_fds[1]) + for fd in self.null_fds + self.save_fds: + os.close(fd)