thanks coderabbit

fiftyone/factory/repo_factory.py

@@ -8,7 +8,7 @@
 
 from pymongo.database import Database
 
-from bson import ObjectId
+import bson
 from typing import Optional
 import fiftyone.core.odm as foo
 from fiftyone.factory.repos.delegated_operation import (
@@ -53,7 +53,7 @@ def delegated_operation_repo() -> DelegatedOperationRepo:
 
     @staticmethod
     def execution_store_repo(
-        dataset_id: Optional[str] = None,
+        dataset_id: Optional[bson.ObjectId] = None,
     ) -> ExecutionStoreRepo:
         """Factory method for execution store repository."""
         if (

fiftyone/operators/store/service.py

@@ -6,6 +6,7 @@
 |
 """
 
+import bson
 import logging
 from typing import Optional, List
 from fiftyone.operators.store.models import StoreDocument, KeyDocument
@@ -14,13 +15,17 @@
 
 
 class ExecutionStoreService:
-    """Service for managing execution store operations."""
-
     def __init__(
         self,
         repo: Optional["ExecutionStoreRepo"] = None,
-        dataset_id: Optional[str] = None,
+        dataset_id: Optional[bson.ObjectId] = None,
     ):
+        """Service for managing execution store operations.
+
+        Args:
+            repo: Optional execution store repository instance
+            dataset_id: Optional bson.ObjectId of the dataset to scope operations to
+        """
         from fiftyone.factory.repo_factory import (
             RepositoryFactory,
             ExecutionStoreRepo,
@@ -134,9 +139,5 @@ def list_keys(self, store_name: str) -> List[str]:
         return self._repo.list_keys(store_name)
 
     def cleanup_for_dataset(self) -> None:
-        """Cleans up the execution store for the specified dataset.
-
-        Args:
-            dataset_id: the ID of the dataset
-        """
+        """Cleans up the execution store for the dataset specified during initialization."""
         self._repo.cleanup_for_dataset()