Add wandb reference

kelvin34501 · Oct 2, 2024 · 761776b · 761776b
1 parent 4258a92
commit 761776b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/omnigibson/utils/profiling_utils.py b/omnigibson/utils/profiling_utils.py
@@ -8,6 +8,7 @@
 import omnigibson.utils.pynvml_utils as pynvml
 
 
+# Method copied from: https://github.com/wandb/wandb/blob/main/wandb/sdk/internal/system/assets/gpu.py
 def gpu_in_use_by_this_process(gpu_handle: "GPUHandle", pid: int) -> bool:
     if psutil is None:
         return False

diff --git a/omnigibson/utils/pynvml_utils.py b/omnigibson/utils/pynvml_utils.py
@@ -1,3 +1,5 @@
+# File copied from: https://github.com/wandb/wandb/blob/main/wandb/vendor/pynvml/pynvml.py
+
 #####
 #
 # Based on nvidia-ml-py version 11.515.48