Update Cloudpickle (#5643)

This commit is contained in:
Richard Liaw
2019-09-09 17:17:29 -07:00
committed by Philipp Moritz
parent ed761900f6
commit 0010f54378
3 changed files with 562 additions and 372 deletions
+1 -1
View File
@@ -2,4 +2,4 @@ from __future__ import absolute_import
from ray.cloudpickle.cloudpickle import *
__version__ = '0.8.0.dev0'
__version__ = '1.2.2.dev0'
File diff suppressed because it is too large Load Diff
+7
View File
@@ -84,6 +84,13 @@ class RNNSpyModel(Model):
capture_index = 0
def _build_layers_v2(self, input_dict, num_outputs, options):
# Previously, a new class object was created during
# deserialization and this `capture_index`
# variable would be refreshed between class instantiations.
# This behavior is no longer the case, so we manually refresh
# the variable.
RNNSpyModel.capture_index = 0
def spy(sequences, state_in, state_out, seq_lens):
if len(sequences) == 1:
return 0 # don't capture inference inputs