Clear module-level globals after workers finish.
()
| 42 | |
| 43 | |
| 44 | def _clear_shared_data(): |
| 45 | """Clear module-level globals after workers finish.""" |
| 46 | global _SHARED_MESSAGES, _SHARED_MODEL_KWARGS |
| 47 | _SHARED_MESSAGES = None |
| 48 | _SHARED_MODEL_KWARGS = None |
| 49 | |
| 50 | |
| 51 | def _worker_fn(args): |