mirror of
https://github.com/microsoft/qlib.git
synced 2026-07-14 08:16:54 +08:00
Update QlibRecorder wrapper
This commit is contained in:
@@ -525,14 +525,33 @@ class QlibRecorder:
|
|||||||
self.get_exp().get_recorder().set_tags(**kwargs)
|
self.get_exp().get_recorder().set_tags(**kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
# error type for reinitialization when starting an experiment
|
||||||
|
class RecorderInitializationError(Exception):
|
||||||
|
def __init__(self, message):
|
||||||
|
super(RecorderInitializationError, self).__init__(message)
|
||||||
|
|
||||||
|
|
||||||
|
class RecorderWrapper(Wrapper):
|
||||||
|
"""
|
||||||
|
Wrapper class for QlibRecorder, which detects whether users reinitialize qlib when already starting an experiment.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def register(self, provider):
|
||||||
|
if self._provider is not None:
|
||||||
|
raise RecorderInitializationError(
|
||||||
|
"Please don't reinitialize Qlib if QlibRecorder is already acivated. Otherwise, the experiment stored location will be modified."
|
||||||
|
)
|
||||||
|
self._provider = provider
|
||||||
|
|
||||||
|
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
if sys.version_info >= (3, 9):
|
if sys.version_info >= (3, 9):
|
||||||
from typing import Annotated
|
from typing import Annotated
|
||||||
|
|
||||||
QlibRecorderWrapper = Annotated[QlibRecorder, Wrapper]
|
QlibRecorderWrapper = Annotated[QlibRecorder, RecorderWrapper]
|
||||||
else:
|
else:
|
||||||
QlibRecorderWrapper = QlibRecorder
|
QlibRecorderWrapper = QlibRecorder
|
||||||
|
|
||||||
# global record
|
# global record
|
||||||
R: QlibRecorderWrapper = Wrapper()
|
R: QlibRecorderWrapper = RecorderWrapper()
|
||||||
|
|||||||
Reference in New Issue
Block a user