Add database_uri as a potential input parameter

ahcore/data/dataset.py

@@ -86,12 +86,10 @@ def __len__(self) -> int:
         return self.cumulative_sizes[-1]
 
     @overload
-    def __getitem__(self, index: int) -> DlupDatasetSample:
-        ...
+    def __getitem__(self, index: int) -> DlupDatasetSample: ...
 
     @overload
-    def __getitem__(self, index: slice) -> list[DlupDatasetSample]:
-        ...
+    def __getitem__(self, index: slice) -> list[DlupDatasetSample]: ...
 
     def __getitem__(self, index: Union[int, slice]) -> DlupDatasetSample | list[DlupDatasetSample]:
         """Returns the sample at the given index."""
@@ -112,6 +110,7 @@ def __init__(
         self,
         data_description: DataDescription,
         pre_transform: Callable[[bool], Callable[[DlupDatasetSample], DlupDatasetSample]],
+        database_uri: str | None = None,
         batch_size: int = 32,  # noqa,pylint: disable=unused-argument
         validate_batch_size: int | None = None,  # noqa,pylint: disable=unused-argument
         num_workers: int = 16,
@@ -129,6 +128,8 @@ def __init__(
             A pre-transform is a callable which is directly applied to the output of the dataset before collation in
             the dataloader. The transforms typically convert the image in the output to a tensor, convert the
             `WsiAnnotations` to a mask or similar.
+        database_uri : str, optional
+            The URI to the database. If not provided, the URI from the data description will be used.
         batch_size : int
             The batch size of the data loader.
         validate_batch_size : int, optional
@@ -160,7 +161,9 @@ def __init__(
         # Data settings
         self.data_description: DataDescription = data_description
 
-        self._data_manager = DataManager(database_uri=data_description.manifest_database_uri)
+        self._data_manager = DataManager(
+            database_uri=data_description.manifest_database_uri if not database_uri else database_uri
+        )
 
         self._batch_size = self.hparams.batch_size  # type: ignore
         self._validate_batch_size = self.hparams.validate_batch_size  # type: ignore