Add return statement

microsoft · Jul 25, 2024 · 8f62cb5 · 8f62cb5
1 parent 62e16d4
commit 8f62cb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hi-ml-azure/src/health_azure/utils.py b/hi-ml-azure/src/health_azure/utils.py
@@ -1180,7 +1180,7 @@ def torch_rank() -> Optional[int]:
         return None
     if distributed.is_available() and distributed.is_initialized():
         return distributed.get_rank()
-
+    return None
 
 def is_local_rank_zero() -> bool:
     """