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
@@ -45,7 +45,7 @@ To use Ray's actor model:
ray.init()
@ray.remote
class Counter():
class Counter(object):
def __init__(self):
self.n = 0
+1 -1
View File
@@ -119,7 +119,7 @@ def no_op(*values):
@ray.remote
class Actor():
class Actor(object):
def ping(self, *values):
pass
+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))
+1 -1
View File
@@ -113,7 +113,7 @@ def dataset_creators(use_cuda):
import torch.optim as optim
class Network():
class Network(object):
def __init__(self, lr=0.01, momentum=0.5):
use_cuda = torch.cuda.is_available()
self.device = device = torch.device("cuda" if use_cuda else "cpu")
+1 -1
View File
@@ -156,7 +156,7 @@ def rollout(model, env):
# given an input, which in our case is an observation.
class Model():
class Model(object):
"""This class holds the neural network weights."""
def __init__(self):
+1 -1
View File
@@ -52,7 +52,7 @@ To use Ray's actor model:
ray.init()
@ray.remote
class Counter():
class Counter(object):
def __init__(self):
self.n = 0
+1 -1
View File
@@ -354,7 +354,7 @@ class NodeLauncher(threading.Thread):
logger.info(prefix + " {}".format(statement))
class ConcurrentCounter():
class ConcurrentCounter(object):
def __init__(self):
self._value = 0
self._lock = threading.Lock()
+1 -1
View File
@@ -8,7 +8,7 @@ import logging
logger = logging.getLogger(__name__)
class LogTimer():
class LogTimer(object):
def __init__(self, message):
self._message = message
@@ -9,7 +9,7 @@ from ray.experimental.serve.metric import MetricMonitor
@pytest.fixture(scope="session")
def start_target_actor(ray_instance):
@ray.remote
class Target():
class Target(object):
def __init__(self):
self.counter_value = 0
+1 -1
View File
@@ -41,7 +41,7 @@ def log_sync_template():
).format(ssh_key=quote(ssh_key))
class NodeSyncMixin():
class NodeSyncMixin(object):
"""Mixin for syncing files to/from a remote dir to a local dir."""
def __init__(self):
+1 -1
View File
@@ -294,7 +294,7 @@ class HyperBandScheduler(FIFOScheduler):
trial_runner.trial_executor.unpause_trial(trial)
class Bracket():
class Bracket(object):
"""Logical object for tracking Hyperband bracket progress. Keeps track
of proper parameters as designated by HyperBand.
+1 -1
View File
@@ -20,7 +20,7 @@ from ray.tune import commands
from ray.tune.result import CONFIG_PREFIX
class Capturing():
class Capturing(object):
def __enter__(self):
self._stdout = sys.stdout
sys.stdout = self._stringio = StringIO()