mirror of
https://github.com/wassname/ray.git
synced 2026-07-18 12:40:56 +08:00
Python 2 compatibility. (#5887)
This commit is contained in:
+1
-1
@@ -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
|
||||
|
||||
|
||||
@@ -119,7 +119,7 @@ def no_op(*values):
|
||||
|
||||
|
||||
@ray.remote
|
||||
class Actor():
|
||||
class Actor(object):
|
||||
def ping(self, *values):
|
||||
pass
|
||||
|
||||
|
||||
@@ -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))
|
||||
|
||||
@@ -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")
|
||||
|
||||
@@ -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):
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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()
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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):
|
||||
|
||||
@@ -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.
|
||||
|
||||
|
||||
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user