mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
46ed3349f3
Summary: Also move the ignores for imports to the bottom in `mypy.ini`, those are much less interesting - start with the stuff people want to work on. Second commit tests the instructions: remove an ignore, fix the issue. Pull Request resolved: https://github.com/pytorch/pytorch/pull/37594 Differential Revision: D21434858 Pulled By: ezyang fbshipit-source-id: 4f1a6868cdb4cb59d072bcf105f48c3a5ba3ff98
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
from __future__ import absolute_import, division, print_function, unicode_literals
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
if sys.version < '3.3':
|
|
# Note(jiayq): in Python 2, FileExistsError is not defined and the
|
|
# error manifests it as OSError.
|
|
FileExistsError = OSError
|
|
|
|
|
|
class FileBaton:
|
|
'''A primitive, file-based synchronization utility.'''
|
|
|
|
def __init__(self, lock_file_path, wait_seconds=0.1):
|
|
'''
|
|
Creates a new :class:`FileBaton`.
|
|
|
|
Args:
|
|
lock_file_path: The path to the file used for locking.
|
|
wait_seconds: The seconds to periorically sleep (spin) when
|
|
calling ``wait()``.
|
|
'''
|
|
self.lock_file_path = lock_file_path
|
|
self.wait_seconds = wait_seconds
|
|
self.fd = None
|
|
|
|
def try_acquire(self):
|
|
'''
|
|
Tries to atomically create a file under exclusive access.
|
|
|
|
Returns:
|
|
True if the file could be created, else False.
|
|
'''
|
|
try:
|
|
self.fd = os.open(self.lock_file_path, os.O_CREAT | os.O_EXCL)
|
|
return True
|
|
except FileExistsError:
|
|
return False
|
|
|
|
def wait(self):
|
|
'''
|
|
Periodically sleeps for a certain amount until the baton is released.
|
|
|
|
The amount of time slept depends on the ``wait_seconds`` parameter
|
|
passed to the constructor.
|
|
'''
|
|
while os.path.exists(self.lock_file_path):
|
|
time.sleep(self.wait_seconds)
|
|
|
|
def release(self):
|
|
'''Releases the baton and removes its file.'''
|
|
if self.fd is not None:
|
|
os.close(self.fd)
|
|
|
|
os.remove(self.lock_file_path)
|