| 1234567891011121314151617181920212223242526 |
- """
- Initialize logging for websocket_proxy tests.
- """
- import sys
- import os
- import logging
- from gpustack.logging import TRACE_LEVEL
- # Ensure local gpustack package is used
- repo_root = os.path.abspath(
- os.path.join(os.path.dirname(__file__), os.pardir, os.pardir)
- )
- if repo_root not in sys.path:
- sys.path.insert(0, repo_root)
- logging.addLevelName(TRACE_LEVEL, "TRACE")
- def trace(self, message, *args, **kwargs):
- if self.isEnabledFor(TRACE_LEVEL):
- self._log(TRACE_LEVEL, message, args, **kwargs)
- logging.Logger.trace = trace
|
