Commit 2a318887 authored by xdrazkov's avatar xdrazkov
Browse files

feat: add abstract debug method to ExperimentLogger and TrainLogger

parent ba4f9598

regstar_library/launch/experiment/loggers/abstract.py

+3 −0
Original line number Diff line number Diff line
@@ -12,6 +12,9 @@ class ExperimentLogger(ABC): 
        self, *args: Any, metadata: dict[str, Any] | None = None, **kwargs: Any

    ) -> None: ...



    @abstractmethod

    def debug(self, msg: str) -> None: ...



    def log_meta_dict(self, metadata: dict[str, Any]) -> None:

        for name, value in metadata.items():

            self.log_meta(name, value)

regstar_library/launch/experiment/loggers/none.py

+3 −0
Original line number Diff line number Diff line
@@ -11,6 +11,9 @@ class NoOpExperimentLogger(ExperimentLogger): 
    ) -> None:

        self._document_id = "noop"



    def debug(self, msg: str) -> None:

        return None



    def log_meta(self, name: str, value: Any) -> None:

        return None

regstar_library/trainer/loggers/abstract.py

+3 −0
Original line number Diff line number Diff line
@@ -27,6 +27,9 @@ class TrainLogger(ABC): 
        self, *args: Any, metadata: dict[str, Any] | None = None, **kwargs: Any

    ) -> None: ...



    @abstractmethod

    def debug(self, msg: str) -> None: ...



    def log_meta_dict(self, metadata: dict[str, Any]) -> None:

        for name, value in metadata.items():

            self.log_meta(name, value)

regstar_library/trainer/loggers/none.py

+3 −0
Original line number Diff line number Diff line
@@ -20,6 +20,9 @@ class NoOpTrainLogger(TrainLogger): 
    ) -> None:

        self._document_id = "noop-run"



    def debug(self, msg: str) -> None:

        return None



    def log_meta(self, name: str, value: Any) -> None:

        return None