Update.

huggingface · Jan 30, 2025 · 7c4a8a5 · 7c4a8a5
1 parent eec2785
commit 7c4a8a5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/data/src/shard_interface.rs b/data/src/shard_interface.rs
@@ -11,7 +11,7 @@ use super::errors::Result;
 pub async fn create_shard_manager(
     shard_storage_config: &StorageConfig,
     download_only_mode: bool,
-) -> Result<Arc<ShardFileManager>> {
+) -> Result<(Arc<ShardFileManager>, Arc<ShardFileManager>)> {
     let shard_session_directory = shard_storage_config
         .staging_directory
         .as_ref()
@@ -30,15 +30,15 @@ pub async fn create_shard_manager(
         .build()
         .await?;
 
-    let session_shard_manager = ShardFileManager::builder(shard_cache_directory)
+    let session_shard_manager = ShardFileManager::builder(shard_session_directory)
         .with_chunk_dedup(!download_only_mode)
         .with_expired_shard_cleanup(false)
         .with_caching(false)
         .with_parent_sfm(cache_shard_manager.clone())
         .build()
         .await?;
 
-    Ok(session_shard_manager)
+    Ok((session_shard_manager, cache_shard_manager))
 }
 
 pub async fn create_shard_client(