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())
