This commit is contained in:
allegroai 2021-02-23 10:38:02 +02:00
parent 60b2f5bd74
commit ae6caf4404
5 changed files with 4 additions and 5 deletions

View File

@ -3,7 +3,6 @@ import hashlib
import time import time
from functools import reduce from functools import reduce
from logging import getLevelName from logging import getLevelName
from multiprocessing import Lock
import attr import attr
import numpy as np import numpy as np

View File

@ -1325,7 +1325,7 @@ class PatchTensorFlowEager(object):
# the operator itself, will do the reporting to the backend # the operator itself, will do the reporting to the backend
py_function( py_function(
_report_summary_op, _report_summary_op,
inp=[writer, step, tag, value, name], Tout=[]) inp=[writer, step, tag, values, name], Tout=[])
except Exception as ex: except Exception as ex:
LoggerRoot.get_base_logger(TensorflowBinding).warning(str(ex)) LoggerRoot.get_base_logger(TensorflowBinding).warning(str(ex))

View File

@ -15,9 +15,8 @@ try:
except ImportError: except ImportError:
pd = None pd = None
from .backend_interface.logger import StdStreamPatch, LogFlusher from .backend_interface.logger import StdStreamPatch
from .backend_interface.task import Task as _Task from .backend_interface.task import Task as _Task
from .backend_interface.task.development.worker import DevWorker
from .backend_interface.task.log import TaskHandler from .backend_interface.task.log import TaskHandler
from .backend_interface.util import mutually_exclusive from .backend_interface.util import mutually_exclusive
from .config import running_remotely, get_cache_dir, config, DEBUG_SIMULATE_REMOTE_TASK from .config import running_remotely, get_cache_dir, config, DEBUG_SIMULATE_REMOTE_TASK

View File

@ -1,3 +1,4 @@
import logging
import os import os
from time import time from time import time
from typing import Optional, AnyStr, IO from typing import Optional, AnyStr, IO

View File

@ -2980,7 +2980,7 @@ class Task(_Task):
# this is so in theory we can close a main task and start a new one # this is so in theory we can close a main task and start a new one
if self.is_main_task(): if self.is_main_task():
Task.__main_task = None Task.__main_task = None
except Exception as ex: except Exception:
# make sure we do not interrupt the exit process # make sure we do not interrupt the exit process
pass pass