Unverified Commit 7b5a1e7d authored by Funtowicz Morgan's avatar Funtowicz Morgan Committed by GitHub
Browse files

Fix importing transformers on Windows (#4997)

parent a9f1fc6c
...@@ -16,7 +16,6 @@ from collections import defaultdict, namedtuple ...@@ -16,7 +16,6 @@ from collections import defaultdict, namedtuple
from datetime import datetime from datetime import datetime
from multiprocessing import Pipe, Process from multiprocessing import Pipe, Process
from multiprocessing.connection import Connection from multiprocessing.connection import Connection
from signal import SIGKILL
from typing import Callable, Iterable, List, NamedTuple, Optional, Union from typing import Callable, Iterable, List, NamedTuple, Optional, Union
from transformers import AutoConfig, PretrainedConfig from transformers import AutoConfig, PretrainedConfig
...@@ -33,6 +32,12 @@ if is_tf_available(): ...@@ -33,6 +32,12 @@ if is_tf_available():
from tensorflow.python.eager import context as tf_context from tensorflow.python.eager import context as tf_context
if platform.system() == "Windows":
from signal import CTRL_C_EVENT as SIGKILL
else:
from signal import SIGKILL
logger = logging.getLogger(__name__) # pylint: disable=invalid-name logger = logging.getLogger(__name__) # pylint: disable=invalid-name
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment