fix typing

This commit is contained in:
Ishaan Jaff 2024-08-02 18:49:16 -07:00
parent f194aa3a93
commit 0463e73ee3

View file

@ -119,14 +119,7 @@ class VertexFineTuningAPI(VertexLLM):
"got response from creating fine tuning job: %s", response.json()
)
vertex_response = ResponseTuningJob(
name=None,
tunedModelDisplayName=None,
baseModel=None,
supervisedTuningSpec=None,
state=None,
createTime=None,
updateTime=None,
vertex_response = ResponseTuningJob( # type: ignore
**response.json(),
)
@ -209,14 +202,7 @@ class VertexFineTuningAPI(VertexLLM):
verbose_logger.debug(
"got response from creating fine tuning job: %s", response.json()
)
vertex_response = ResponseTuningJob(
name=None,
tunedModelDisplayName=None,
baseModel=None,
supervisedTuningSpec=None,
state=None,
createTime=None,
updateTime=None,
vertex_response = ResponseTuningJob( # type: ignore
**response.json(),
)