Skip to content

Commit 992daf4

Browse files
committed
only change if its a string
1 parent dc62cb3 commit 992daf4

File tree

1 file changed

+4
-2
lines changed
  • src/india_api/internal/inputs/indiadb

1 file changed

+4
-2
lines changed

src/india_api/internal/inputs/indiadb/client.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -272,7 +272,8 @@ def get_site_forecast(self, site_uuid: str, email:str) -> list[internal.Predicte
272272
with self._get_session() as session:
273273
check_user_has_access_to_site(session=session, email=email, site_uuid=site_uuid)
274274

275-
site_uuid = UUID(site_uuid)
275+
if isinstance(site_uuid, str):
276+
site_uuid = UUID(site_uuid)
276277

277278
values = get_latest_forecast_values_by_site(
278279
session,
@@ -306,7 +307,8 @@ def get_site_generation(self, site_uuid: str, email:str) -> list[internal.Actual
306307
with self._get_session() as session:
307308
check_user_has_access_to_site(session=session, email=email, site_uuid=site_uuid)
308309

309-
site_uuid = UUID(site_uuid)
310+
if isinstance(site_uuid, str):
311+
site_uuid = UUID(site_uuid)
310312

311313
# read actual generations
312314
values = get_pv_generation_by_sites(

0 commit comments

Comments
 (0)