Commit 5ab89843 authored by Yuxin Wu's avatar Yuxin Wu

os==windows checking in zmq

parent 273e6f91
...@@ -6,6 +6,7 @@ ...@@ -6,6 +6,7 @@
import tensorflow as tf import tensorflow as tf
import multiprocessing as mp import multiprocessing as mp
import time import time
import os
import threading import threading
from abc import abstractmethod, ABCMeta from abc import abstractmethod, ABCMeta
from collections import defaultdict from collections import defaultdict
...@@ -105,6 +106,7 @@ class SimulatorMaster(threading.Thread): ...@@ -105,6 +106,7 @@ class SimulatorMaster(threading.Thread):
def __init__(self, pipe_c2s, pipe_s2c): def __init__(self, pipe_c2s, pipe_s2c):
super(SimulatorMaster, self).__init__() super(SimulatorMaster, self).__init__()
assert os.name != 'nt', "Doesn't support windows!"
self.daemon = True self.daemon = True
self.name = 'SimulatorMaster' self.name = 'SimulatorMaster'
......
...@@ -119,6 +119,7 @@ class PrefetchDataZMQ(ProxyDataFlow): ...@@ -119,6 +119,7 @@ class PrefetchDataZMQ(ProxyDataFlow):
nr_proc (int): number of processes to use. nr_proc (int): number of processes to use.
hwm (int): the zmq "high-water mark" for both sender and receiver. hwm (int): the zmq "high-water mark" for both sender and receiver.
""" """
assert os.name != 'nt', "PrefetchDataZMQ doesn't support windows! Consider PrefetchData instead."
super(PrefetchDataZMQ, self).__init__(ds) super(PrefetchDataZMQ, self).__init__(ds)
try: try:
self._size = ds.size() self._size = ds.size()
......
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