fix pyright check

bioimage-io · Mar 3, 2024 · 3b9e6f2 · 3b9e6f2
1 parent e0370f2
commit 3b9e6f2
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_backoffice.py b/tests/test_backoffice.py
@@ -1,13 +1,14 @@
 import os
 
 from backoffice import BackOffice
+from backoffice.s3_structure.versions import StageNr
 
 
 def test_backoffice(
     backoffice: BackOffice, package_url: str, package_id: str, s3_test_folder_url: str
 ):
     backoffice.stage(resource_id=package_id, package_url=package_url)
-    backoffice.test(resource_id=package_id, stage_nr=1)
-    backoffice.await_review(resource_id=package_id, stage_nr=1)
-    backoffice.publish(resource_id=package_id, stage_nr=1)
+    backoffice.test(resource_id=package_id, stage_nr=StageNr(1))
+    backoffice.await_review(resource_id=package_id, stage_nr=StageNr(1))
+    backoffice.publish(resource_id=package_id, stage_nr=StageNr(1))
     backoffice.backup(os.environ["ZENODO_TEST_URL"])