modify data sync result method

AllenInstitute · Nov 12, 2024 · 6ff280f · 6ff280f
1 parent 084c428
commit 6ff280f
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/aibs_informatics_core/models/data_sync.py b/src/aibs_informatics_core/models/data_sync.py
@@ -201,13 +201,13 @@ class BatchDataSyncResult(DataSyncResult):
     successful_requests_count: int = 0
     failed_requests_count: int = 0
 
-    def increment_successful_requests_count(self) -> None:
-        self.successful_requests_count += 1
-        self.total_requests_count += 1
+    def increment_successful_requests_count(self, increment: int = 1) -> None:
+        self.successful_requests_count += increment
+        self.total_requests_count += increment
 
-    def increment_failed_requests_count(self) -> None:
-        self.failed_requests_count += 1
-        self.total_requests_count += 1
+    def increment_failed_requests_count(self, increment: int = 1) -> None:
+        self.failed_requests_count += increment
+        self.total_requests_count += increment
 
 
 @dataclass