diff --git a/google/cloud/aiplatform/metadata/experiment_run_resource.py b/google/cloud/aiplatform/metadata/experiment_run_resource.py index 055bd8d981..39ef0eaa9d 100644 --- a/google/cloud/aiplatform/metadata/experiment_run_resource.py +++ b/google/cloud/aiplatform/metadata/experiment_run_resource.py @@ -939,7 +939,7 @@ def log_params(self, params: Dict[str, Union[float, int, str]]): Required. Parameter key/value pairs. Raises: - ValueError: If key is not str or value is not float, int, str. + TypeError: If key is not str or value is not float, int, str. """ # query the latest run execution resource before logging. for key, value in params.items(): @@ -968,7 +968,7 @@ def log_metrics(self, metrics: Dict[str, Union[float, int, str]]): ``` Args: - metrics (Dict[str, Union[float, int]]): + metrics (Dict[str, Union[float, int, str]]): Required. Metrics key/value pairs. Raises: TypeError: If keys are not str or values are not float, int, or str.