Skip to content

Commit

Permalink
feat: print versions of 1st party dependencies to worker log
Browse files Browse the repository at this point in the history
When diagnosing issues it can be useful to have the versions of the
`deadline` and `openjd-sessions` packages available in the log. This
commit adds that information.

Signed-off-by: Daniel Neilson <53624638+ddneilson@users.noreply.github.com>
  • Loading branch information
ddneilson committed Nov 7, 2023
1 parent 60959c0 commit b2ac38e
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 2 deletions.
11 changes: 9 additions & 2 deletions src/deadline_worker_agent/startup/entrypoint.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,10 @@
import shutil
from pathlib import Path

from openjd.model import version as openjd_model_version
from openjd.sessions import version as openjd_sessions_version
from openjd.sessions import LOG as OPENJD_SESSION_LOG
from deadline.job_attachments import version as deadline_job_attach_version
from pydantic import PositiveFloat

from .._version import __version__
Expand Down Expand Up @@ -331,11 +334,15 @@ def _remove_logging_handler(handler: logging.Handler) -> None:


def _log_agent_info() -> None:
_logger.info("Agent Version: %s", __version__)
_logger.info("Installed at: %s", str(Path(__file__).resolve().parent.parent))
_logger.info(f"Python Interpreter: {sys.executable}")
_logger.info("Python Version: %s", sys.version.replace("\n", " - "))
_logger.info(f"Platform: {sys.platform}")
_logger.info("Agent Version: %s", __version__)
_logger.info("Installed at: %s", str(Path(__file__).resolve().parent.parent))
_logger.info("Dependency versions installed:")
_logger.info("\topenjd.model: %s", openjd_model_version)
_logger.info("\topenjd.sessions: %s", openjd_sessions_version)
_logger.info("\tdeadline.job_attachments: %s", deadline_job_attach_version)


def _get_gpu_count(*, verbose: bool = True) -> int:
Expand Down
4 changes: 4 additions & 0 deletions test/unit/scheduler/test_session_cleanup.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,10 @@ class FakeSessionUser(SessionUser):
def __init__(self, user: str):
self.user = user

@staticmethod
def get_process_user() -> str:
return ""


class TestSessionUserCleanupManager:
@pytest.fixture
Expand Down

0 comments on commit b2ac38e

Please sign in to comment.