示例#1
0
 def communicate_run_start(self,
                           run_pb: pb.RunRecord) -> Optional[pb.Result]:
     run_start = pb.RunStartRequest()
     run_start.run.CopyFrom(run_pb)
     rec = self._make_request(run_start=run_start)
     result = self._communicate(rec)
     return result
示例#2
0
    def fn(write_function, logdir="./", save=True, root_dir="./"):

        with backend_interface() as interface:
            proto_run = pb.RunRecord()
            mocked_run._make_proto_run(proto_run)

            run_start = pb.RunStartRequest()
            run_start.run.CopyFrom(proto_run)

            request = pb.Request()
            request.run_start.CopyFrom(run_start)

            record = pb.Record()
            record.request.CopyFrom(request)
            internal_hm.handle_request_run_start(record)
            internal_hm._tb_watcher.add(logdir, save, root_dir)

            # need to sleep to give time for the tb_watcher delay
            time.sleep(15)
            write_function()

        ctx_util = parse_ctx(mock_server.ctx)
        return ctx_util
示例#3
0
 def communicate_run_start(self):
     run_start = wandb_internal_pb2.RunStartRequest()
     rec = self._make_request(run_start=run_start)
     result = self._communicate(rec)
     return result
示例#4
0
 def communicate_run_start(self, run_pb):
     run_start = pb.RunStartRequest()
     run_start.run.CopyFrom(run_pb)
     rec = self._make_request(run_start=run_start)
     result = self._communicate(rec)
     return result
示例#5
0
 def communicate_run_start(self, run_pb: pb.RunRecord) -> bool:
     run_start = pb.RunStartRequest()
     run_start.run.CopyFrom(run_pb)
     result = self._communicate_run_start(run_start)
     return result is not None