Update requirements

actorch/algorithms/algorithm.py

@@ -43,7 +43,7 @@
 import torch.nn.functional as F
 from gym import Env, spaces
 from numpy import ndarray
-from ray.train.session import world_rank
+from ray.train import world_rank
 from ray.train.torch import accelerate, prepare_data_loader
 from ray.tune import Trainable
 from ray.tune import result as tune_result

actorch/utils.py

@@ -28,7 +28,7 @@
 import GPUtil
 import psutil
 import yaml
-from ray.tune.logger import SafeFallbackEncoder
+from ray.tune.utils.util import SafeFallbackEncoder
 
 
 __all__ = [

requirements/requirements.txt

@@ -5,7 +5,7 @@ gym>=0.25.0
 numpy>=1.19.5
 psutil>=5.7.0
 PyYAML>=5.3.1
-ray[tune]>=1.13.0
+ray[tune]>=1.13.0,<2.*
 scipy>=1.4.1
 tensorboard>=2.6.0
 -f https://download.pytorch.org/whl/torch_stable.html  # PyTorch installation on Windows