run.py 624 B

123456789101112131415161718192021222324252627
  1. #
  2. """InternalRun - Internal-only run object.
  3. Semi-stubbed run for internal process use.
  4. """
  5. from typing_extensions import override
  6. from wandb.sdk import wandb_run
  7. class InternalRun(wandb_run.Run):
  8. def __init__(self, run_obj, settings, datatypes_cb):
  9. super().__init__(settings=settings)
  10. self._run_obj = run_obj
  11. self._datatypes_cb = datatypes_cb
  12. @override
  13. def _set_backend(self, backend):
  14. # This type of run object can't have a backend
  15. # or do any writes.
  16. pass
  17. @override
  18. def _publish_file(self, fname: str) -> None:
  19. self._datatypes_cb(fname)