Fix mypy errors in Optuna 4.0+

This commit is contained in:
c-bata
2024-09-11 16:32:38 +09:00
parent 7d6b619212
commit 5a1d6365cd
4 changed files with 27 additions and 9 deletions
+6 -2
View File
@@ -539,7 +539,9 @@ class APITestCase(TestCase):
with tempfile.NamedTemporaryFile() as f:
f.write(b"dummy")
f.flush()
artifact_id = upload_artifact(study, f.name, artifact_store)
artifact_id = upload_artifact(
study_or_trial=study, file_path=f.name, artifact_store=artifact_store
)
app = create_app(storage, artifact_store)
@@ -575,7 +577,9 @@ class APITestCase(TestCase):
with tempfile.NamedTemporaryFile() as f:
f.write(b"dummy")
f.flush()
artifact_id = upload_artifact(study, f.name, artifact_store)
artifact_id = upload_artifact(
study_or_trial=study, file_path=f.name, artifact_store=artifact_store
)
app = create_app(storage, artifact_store)