Skip to content

Commit 67edb48

Browse files
mahlau-flexyaugenst-flex
authored andcommitted
REFC /web: updated isolated components to pydantic v2
1 parent f1ebdef commit 67edb48

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

tidy3d/web/api/material_fitter.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
import numpy as np
1212
import requests
13-
from pydantic.v1 import BaseModel, Field
13+
from pydantic import BaseModel, Field
1414

1515
from tidy3d.plugins.dispersion import DispersionFitter
1616
from tidy3d.web.core.http_util import http
@@ -95,10 +95,10 @@ def submit(cls, fitter: DispersionFitter, options: FitterOptions) -> MaterialFit
9595
raise resp.raise_for_status()
9696
fitter_req = _FitterRequest(
9797
fileName=os.path.basename(temp.name),
98-
jsonInput=options.json(exclude_none=True),
98+
jsonInput=options.model_dump_json(exclude_none=True),
9999
resourcePath=uid,
100100
)
101-
resp = http.post("tidy3d/fitter/fit", json=fitter_req.dict())
101+
resp = http.post("tidy3d/fitter/fit", json=fitter_req.model_dump())
102102
return cls(dispersion_fitter=fitter, **resp)
103103

104104
def sync_status(self) -> None:

tidy3d/web/core/s3utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313

1414
import boto3
1515
from boto3.s3.transfer import TransferConfig
16-
from pydantic.v1 import BaseModel, Field
16+
from pydantic import BaseModel, Field
1717
from rich.progress import (
1818
BarColumn,
1919
DownloadColumn,
@@ -208,7 +208,7 @@ def get_s3_sts_token(
208208
if extra_arguments is not None:
209209
method += "&" + "&".join(f"{k}={v}" for k, v in extra_arguments.items())
210210
resp = http.get(method)
211-
token = _S3STSToken.parse_obj(resp)
211+
token = _S3STSToken.model_validate(resp)
212212
_s3_sts_tokens[cache_key] = token
213213
return _s3_sts_tokens[cache_key]
214214

0 commit comments

Comments
 (0)