From 53378170e07441e0bb6c9dbb3581bf6be289936f Mon Sep 17 00:00:00 2001
From: Farzan Taj <farzantaj@outlook.com>
Date: Thu, 17 Dec 2020 22:17:08 -0500
Subject: [PATCH] [tune] Change pickle to ray.cloudpickle -- support large
 models (#12958)

* Change pickle to ray.cloudpickle

* Change pickle import to ray.cloudpickle
---
 python/ray/tune/trainable.py       | 2 +-
 python/ray/tune/utils/trainable.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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