wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 04:07:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #152 from amplab/fixnumpyscalar

Browse Source 
Fix serialization of numpy scalars, implement more numpy types, empty arrays
This commit is contained in:
Robert Nishihara
2016-06-23 18:42:08 -07:00
committed by GitHub
parent 160bd6ed4f 5ecc2ab67d
commit fde802fb45
4 changed files with 89 additions and 112 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/python/ray/worker.py
+3
View File
@@ -70,6 +70,9 @@ class Worker(object):
result = serialization.Str(result)
elif isinstance(result, np.ndarray):
result = result.view(serialization.NDArray)
elif isinstance(result, np.generic):
return result
# TODO(pcm): close the associated memory segment; if we don't, this leaks memory (but very little, so it is ok for now)
elif result == None:
return None # can't subclass None and don't need to because there is a global None
# TODO(pcm): close the associated memory segment; if we don't, this leaks memory (but very little, so it is ok for now)