Add kubernetes runtime environment check in usage logging

Signed-off-by: Niels Nuyttens <niels@nannyml.com>
NannyML · Jul 3, 2023 · e400726 · e400726
1 parent 5560a10
commit e400726
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/nannyml/usage_logging.py b/nannyml/usage_logging.py
@@ -266,6 +266,10 @@ def _is_running_in_docker():
  return False
 
 
+def is_running_in_kubernetes():
+ return Path('/var/run/secrets/kubernetes.io/').exists()
+
+
 # Inspired by
 # https://github.com/zenml-io/zenml/blob/275109da08b783d5d2cd508b5f703aed0c66e485/src/zenml/environment.py#L182
 # and https://stackoverflow.com/a/39662359