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

os==windows checking in zmq

parent 273e6f91
......@@ -6,6 +6,7 @@
import tensorflow as tf
import multiprocessing as mp
import time
import os
import threading
from abc import abstractmethod, ABCMeta
from collections import defaultdict
......@@ -105,6 +106,7 @@ class SimulatorMaster(threading.Thread):
def __init__(self, pipe_c2s, pipe_s2c):
super(SimulatorMaster, self).__init__()
assert os.name != 'nt', "Doesn't support windows!"
self.daemon = True
self.name = 'SimulatorMaster'
......
......@@ -119,6 +119,7 @@ class PrefetchDataZMQ(ProxyDataFlow):
nr_proc (int): number of processes to use.
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)
try:
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