fix pyfunc multiprocessing serialize problem

2021-09-28 19:24:50 +08:00 · 2021-09-28 19:24:50 +08:00 · e8c5e4244c
parent 36c591b113
commit e8c5e4244c
1 changed files with 3 additions and 0 deletions
--- a/mindspore/dataset/engine/datasets.py
+++ b/mindspore/dataset/engine/datasets.py
@ -2590,6 +2590,9 @@ class _PythonCallable:
        # Invoke original Python callable in master process in case the pool is gone.
        return self.py_callable(*args)

+    def to_json(self):
+        return self.py_callable.to_json()
+
    def _pool_is_running(self):
        # note here: the RUN state of python3.7 and python3.8 is different:
        # python3.7: RUN = 0