Skip to content

Commit 4e96106

Browse files
authored
Merge pull request #35 from atomic-data-sciences/bugfix/new_data_types
Add radial scan support to client
2 parents 82c7eaf + 2ae4c3d commit 4e96106

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/atomicds/client.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,9 @@ def get(
173173
def _get_result_data(
174174
self,
175175
data_id: str,
176-
data_type: Literal["xps", "rheed_image", "rheed_stationary", "rheed_rotating"],
176+
data_type: Literal[
177+
"xps", "rheed_image", "rheed_stationary", "rheed_rotating", "rheed_xscan"
178+
],
177179
) -> RHEEDVideoResult | RHEEDImageResult | XPSResult | None:
178180
if data_type == "xps":
179181
result: dict = self._get(sub_url=f"xps/{data_id}") # type: ignore # noqa: PGH003
@@ -191,7 +193,7 @@ def _get_result_data(
191193
if data_type == "rheed_image":
192194
return self._get_rheed_image_result(data_id)
193195

194-
if data_type in ["rheed_stationary", "rheed_rotating"]:
196+
if data_type in ["rheed_stationary", "rheed_rotating", "rheed_xscan"]:
195197
# Get timeseries data
196198
timeseries_data = self._get_rheed_timeseries_result(data_id)
197199

0 commit comments

Comments
 (0)