diff --git a/grain/_src/python/grain_pool.py b/grain/_src/python/grain_pool.py index 0bcf381e3..1f731ca3e 100644 --- a/grain/_src/python/grain_pool.py +++ b/grain/_src/python/grain_pool.py @@ -419,6 +419,9 @@ def __init__( ), ), ) + print(f"worker_index: {worker_index}") + print(f"worker_count: {options.num_workers}") + print(f"enable_profiling: {options.enable_profiling}") # The process kwargs must all be pickable and will be unpickle before # absl.app.run() is called. We send arguments via a queue to ensure that # they are unpickled after absl.app.run() was called in the child