Merge pull request #51 from UKGovernmentBEIS/craig/trace-public

Use public `trace_action`/`trace_message` functions
UKGovernmentBEIS · Jan 10, 2025 · a9dbd13 · a9dbd13
2 parents 6ee9fbf + 0a0cd6b
commit a9dbd13
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 9 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -10,7 +10,7 @@ packages = [
 
 [tool.poetry.dependencies]
 python = "^3.10"
-inspect-ai = ">=0.3.54"
+inspect-ai = ">=0.3.57"
 kubernetes = "^31.0.0"
 
 [tool.poetry.group.dev.dependencies]

diff --git a/src/k8s_sandbox/_logger.py b/src/k8s_sandbox/_logger.py
@@ -4,8 +4,7 @@
 from contextlib import contextmanager
 from typing import Any, Generator
 
-# TODO: Accessing private functions. To be made public by Inspect.
-from inspect_ai._util.trace import trace_action, trace_message
+from inspect_ai.util import trace_action, trace_message
 
 logger = logging.getLogger(__name__)