def create_experiment(self): """Creates an Experiment for this upload session and returns the ID.""" logger.info("Creating experiment") request = write_service_pb2.CreateExperimentRequest() response = grpc_util.call_with_retries(self._api.CreateExperiment, request) self._request_builder = _RequestBuilder(response.experiment_id) return response.experiment_id
def create_experiment(self): """Creates an Experiment for this upload session and returns the ID.""" logger.info("Creating experiment") request = write_service_pb2.CreateExperimentRequest( name=self._name, description=self._description) response = grpc_util.call_with_retries(self._api.CreateExperiment, request) self._request_sender = _BatchedRequestSender(response.experiment_id, self._api, self._rpc_rate_limiter) return response.experiment_id
def test_create_experiment_with_name(self): logdir = "/logs/foo" mock_client = _create_mock_client() new_name = "This is the new name" uploader = _create_uploader(mock_client, logdir, name=new_name) eid = uploader.create_experiment() self.assertEqual(eid, "123") mock_client.CreateExperiment.assert_called_once() (args, _) = mock_client.CreateExperiment.call_args expected_request = write_service_pb2.CreateExperimentRequest( name=new_name, ) self.assertEqual(args[0], expected_request)
def test_create_experiment_with_description(self): logdir = "/logs/foo" mock_client = _create_mock_client() new_description = """ **description**" may have "strange" unicode chars 🌴 \\/<> """ uploader = uploader_lib.TensorBoardUploader( mock_client, logdir, description=new_description) eid = uploader.create_experiment() self.assertEqual(eid, "123") mock_client.CreateExperiment.assert_called_once() (args, _) = mock_client.CreateExperiment.call_args expected_request = write_service_pb2.CreateExperimentRequest( description=new_description, ) self.assertEqual(args[0], expected_request)
def create_experiment(self): """Creates an Experiment for this upload session and returns the ID.""" logger.info("Creating experiment") request = write_service_pb2.CreateExperimentRequest( name=self._name, description=self._description) response = grpc_util.call_with_retries(self._api.CreateExperiment, request) self._tracker = upload_tracker.UploadTracker(verbosity=self._verbosity) self._request_sender = _BatchedRequestSender( response.experiment_id, self._api, allowed_plugins=self._allowed_plugins, upload_limits=self._upload_limits, rpc_rate_limiter=self._rpc_rate_limiter, tensor_rpc_rate_limiter=self._tensor_rpc_rate_limiter, blob_rpc_rate_limiter=self._blob_rpc_rate_limiter, tracker=self._tracker, ) return response.experiment_id
def test_create_experiment_with_all_metadata(self): logdir = "/logs/foo" mock_client = _create_mock_client() new_description = """ **description**" may have "strange" unicode chars 🌴 \/<> """ new_name = "This is a cool name." uploader = _create_uploader( mock_client, logdir, name=new_name, description=new_description ) eid = uploader.create_experiment() self.assertEqual(eid, "123") mock_client.CreateExperiment.assert_called_once() (args, _) = mock_client.CreateExperiment.call_args expected_request = write_service_pb2.CreateExperimentRequest( name=new_name, description=new_description, ) self.assertEqual(args[0], expected_request)
def testCreateExperiment(self): self._stub.CreateExperiment( write_service_pb2.CreateExperimentRequest())