Update type assertions in mutate_load_script()

simonsobs · May 7, 2024 · 4c7a439 · 4c7a439
1 parent 02e5ca0
commit 4c7a439
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/nextline_schedule/schema/mutation.py b/src/nextline_schedule/schema/mutation.py
@@ -33,10 +33,12 @@ def update(self, info: Info, input: MutationSchedulerInput) -> bool:
 
 
 async def mutate_load_script(info: Info) -> bool:
-    nextline: Nextline = info.context["nextline"]
+    nextline = info.context["nextline"]
+    assert isinstance(nextline, Nextline)
     scheduler = info.context['schedule']['scheduler']
-    assert isinstance(scheduler, RequestStatement)
+    assert callable(scheduler)
     statement = await scheduler()
+    assert isinstance(statement, str)
     await nextline.reset(statement=statement)
     return True