Python 2 compatibility. (#5887)

This commit is contained in:
Robert Nishihara
2019-10-10 19:09:25 -07:00
committed by GitHub
parent c3b2ae26c5
commit 523c764c25
12 changed files with 12 additions and 12 deletions
+1 -1
View File
@@ -48,7 +48,7 @@ def random_one_hot_labels(shape):
# Use GPU wth
# @ray.remote(num_gpus=1)
@ray.remote
class Network():
class Network(object):
def __init__(self):
self.model = create_keras_model()
self.dataset = np.random.random((1000, 32))