Fix logging of GPU ID

[VibhuJawa]

This PR fixes: #55.

For context/understanding see below:

from distributed import get_worker
def get_worker_name():
    worker = get_worker()  
    return worker.name
    

# Submit the function to all workers
futures = [client.submit(get_worker_name, pure=False) for _ in range(0,10)]
worker_names = client.gather(futures)
print(worker_names)

[3, 0, 6, 3, 5, 2, 4, 7, 3, 1]