diff --git a/python/ray/tune/trainable.py b/python/ray/tune/trainable.py index aa08fd03b..c2915b19f 100644 --- a/python/ray/tune/trainable.py +++ b/python/ray/tune/trainable.py @@ -5,7 +5,7 @@ from datetime import datetime import copy import logging import os -import pickle +import ray.cloudpickle as pickle import platform from ray.tune.utils.trainable import TrainableUtil diff --git a/python/ray/tune/utils/trainable.py b/python/ray/tune/utils/trainable.py index bb9299793..cc61e3a83 100644 --- a/python/ray/tune/utils/trainable.py +++ b/python/ray/tune/utils/trainable.py @@ -5,7 +5,7 @@ import shutil from typing import Dict, Any import pandas as pd -import pickle +import ray.cloudpickle as pickle import os import ray